利用TensorFlow构建一个简单神经元

最新推荐文章于 2024-06-16 09:57:09 发布

阿瓜的blog

最新推荐文章于 2024-06-16 09:57:09 发布

阅读量301

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_46108138/article/details/112300545

这篇博客介绍了如何使用TensorFlow库构建一个简单的神经网络模型来预测房价。通过创建一个全连接层（Dense）并配置优化器（sgd）和损失函数（mean_squared_error），模型在给定的训练数据集上进行了训练。输入数据包括x值，对应的y值为房价。最终，模型对x=7的房价进行了预测。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

建立一个简单的神经元进行房价预测

import tensorflow as tf
import numpy as np
from tensorflow import keras
model =tf.keras.Sequential([keras.layers.Dense(units=1,input_shape=[1])])#建立一个
model.compile(optimizer='sgd',loss='mean_squared_error')
xs=np.array([1.0,2.0,3.0,4.0,5.0,6.0],dtype=float)
ys=np.array([1.0,1.5,2.0,2.5,3.0,3.5],dtype=float)
model.fit(xs,ys,epochs=5000)
print(model.predict([7.0]))