第一个Keras程序

最新推荐文章于 2021-07-03 18:58:21 发布

Go 鹏ya

最新推荐文章于 2021-07-03 18:58:21 发布

阅读量158

点赞数

分类专栏： keras

本文链接：https://blog.csdn.net/qq_42961603/article/details/115451758

版权

keras 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

这个博客展示了如何使用Keras库构建并训练一个简单的神经网络模型。通过`Sequential`模型和`Dense`层，作者编译了模型并用SGD优化器和MSE损失函数进行训练。在3000步的训练过程中，每500步打印一次成本。最后，模型用于预测并绘制预测结果。

摘要由CSDN通过智能技术生成

第一个Keras程序

import keras
import numpy as np
import matplotlib.pyplot as plt
from keras.models import Sequential
from keras.layers import Dense

model.compile(optimizer='sgd',loss='mse')
for step in range(3000):
	cost = model.train_on_batch(x_data,y_data)
	if step%500==0:
		print('cost:',cost)
W,b = model.layers[0].get_weights()
print('W:',W,'b:',b)

y_pred = model.predict(x_data)
plt.plot(x_data,y_pred,'r-',lw=3)
plt.show()