TensorFlow 示例平方米转亩（二）

李建军

于 2025-02-08 07:04:11 发布

阅读量359

点赞数 5

分类专栏： TensorFlow 文章标签： tensorflow 人工智能 python

本文链接：https://blog.youkuaiyun.com/hb_ljj/article/details/145506098

版权

一训练结果

二完整代码

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.callbacks import EarlyStopping

# === 中文字体设置 ===
plt.rcParams['font.sans-serif'] = ['Microsoft YaHei']
plt.rcParams['axes.unicode_minus'] = False

# 1. 数据归一化（统一比例）
平方米 = np.array([15, 30, 50, 100, 150, 300, 315, 666.67, 1000], dtype=float)
亩 = np.array([0.0225, 0.0450, 0.0750, 0.1500, 0.2250, 0.4500, 0.4725, 1.0000, 1.5000], dtype=float)

max_value = 1000.0  # 输入最大值
平方米_normalized = 平方米.reshape(-1, 1) / max_value  # 输入归一化到 [0,1]
亩_normalized = 亩 / (0.0015 * max_value)             # 输出归一化到 [0,1]

# 2. 构建模型（禁用偏置项）
model = tf.keras.Sequential([
    tf.keras.layers.Dense(1, input_shape=[1], use_bias&#