import numpy as np import matplotlib.pyplot as plt # 生成sin函数数据 import pip import pydot x = np.arange(0, 2*np.pi, 0.1) y = np.sin(x) # 可视化sin函数 plt.plot(x, y) plt.show() from keras.models import Sequential from keras.layers import Dense, SimpleRNN # 准备数据 dataX, dataY = [], [] for i in range(len(y)-1): dataX.append(y[i:i+1]) dataY.append(y[i+1]) dataX = np.array(dataX) dataY = np.array(dataY) # 划分训练集和测试集 train_size = int(len(dataY) * 0.7) test_size = len(dataY) - train_size trainX, testX = np.array(dataX[0:train_size]), np.array(dataX[train_size:len(dataX)]) trainY, testY = np.array(dataY[0:train_size]), np.array(dataY[train_size:len(dataY)]) # 调整输入数据的形状 trainX = np.reshape(trainX, (trainX.shape[0], 1, trainX.shape[1])) testX = np.reshape(testX, (testX.shape[0], 1, testX.shape[1])) # 定义模型结构 model = Sequential() model.add(SimpleRNN(units=10, input_shape=(1, 1))) model.add(Dense(units=1)) # 编译模型 model.compile(optimizer='adam', loss='mse') # 训练模型 history = model.fit(trainX, trainY, epochs=100, validation_data=(testX, testY)) # 可视化损失函数 plt.plot(history.history['loss']) plt.plot(history.history['val_loss']) plt.title('Model Loss') plt.ylabel('Loss') plt.xlabel('Epoch') plt.legend(['Train', 'Test'], loc='upper right') plt.show() #预测结果 trainPredict = model.predict(trainX) testPredict = model.predict(testX) # 可视化预测结果 plt.plot(y) plt.plot(np.concatenate((trainPredict, testPredict))) plt.show()模型可视化
时间: 2024-03-25 08:39:25 浏览: 69
import numpy as np(2).py
要可视化 Keras 模型,可以使用 pydot 和 Graphviz 库来创建模型的可视化图。以下是一个简单的例子:
```
from keras.utils.vis_utils import plot_model
# 定义模型结构
model = Sequential()
model.add(SimpleRNN(units=10, input_shape=(1, 1)))
model.add(Dense(units=1))
# 可视化模型
plot_model(model, to_file='model.png', show_shapes=True)
```
该代码将创建一个名为 `model.png` 的文件,其中包含了模型的可视化图。需要注意的是,需要先安装 pydot 和 Graphviz 库,才能使用 `plot_model` 函数。可以使用以下命令安装这两个库:
```
pip install pydot
pip install graphviz
```
如果在 Windows 系统上使用,还需要手动安装 Graphviz 软件,并将其路径添加到系统环境变量中。
阅读全文