matplotlib 动态显示训练过程中的数据和模子的决策边界
Github[*]https://github.com/matplotlib/matplotlib
官网
[*]https://matplotlib.org/stable/
文档
[*]https://matplotlib.org/stable/api/index.html
简介
matplotlib 是 Python 中最常用的绘图库之一，用于创建各种类型的静态、动态和交互式可视化。
动态显示训练过程中的数据和模子的决策边界
https://img-blog.csdnimg.cn/direct/a0601c353cf148dd97bd99f5980c8b3f.gif#pic_center
安装
pip install tensorflow==2.13.1
pip install matplotlib==3.7.5
pip install numpy==1.24.3
源码
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
# 生成数据
np.random.seed(0)
num_samples_per_class = 500
negative_samples = np.random.multivariate_normal(
mean=,
cov=[, ],
size=num_samples_per_class
)
positive_samples = np.random.multivariate_normal(
mean=,
cov=[, ],
size=num_samples_per_class
)
inputs = np.vstack((negative_samples, positive_samples)).astype(np.float32)
targets = np.vstack((np.zeros((num_samples_per_class, 1)), np.ones((num_samples_per_class, 1)))).astype(np.float32)
# 将数据分为训练集和测试集
train_size = int(0.8 * len(inputs))
X_train, X_test = inputs[:train_size], inputs
y_train, y_test = targets[:train_size], targets
# 构建二分类模型
model = Sequential([
# 输入层：输入形状为 (2,)
# 第一个隐藏层：包含 4 个节点，激活函数使用 ReLU
Dense(4, activation='relu', input_shape=(2,)),
# 输出层：包含 1 个节点，激活函数使用 Sigmoid（因为是二分类问题）
Dense(1, activation='sigmoid')
])
# 编译模型
# 指定优化器为 Adam，损失函数为二分类交叉熵，评估指标为准确率
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
# 准备绘图
fig, ax = plt.subplots()
cmap_light = ListedColormap(['#FFAAAA', '#AAAAFF'])
cmap_bold = ListedColormap(['#FF0000', '#0000FF'])
# 动态绘制函数
def plot_decision_boundary(epoch, logs):
ax.clear()
x_min, x_max = X_train[:, 0].min() - 1, X_train[:, 0].max() + 1
y_min, y_max = X_train[:, 1].min() - 1, X_train[:, 1].max() + 1
xx, yy = np.meshgrid(np.arange(x_min, x_max, 0.1),
np.arange(y_min, y_max, 0.1))
grid = np.c_
probs = model.predict(grid).reshape(xx.shape)
ax.contourf(xx, yy, probs, alpha=0.8, cmap=cmap_light)
ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train[:, 0], edgecolor='k', cmap=cmap_bold)
ax.set_title(f'Epoch {epoch+1}')
plt.draw()
plt.pause(0.01)
# 自定义回调函数
class PlotCallback(tf.keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs=None):
plot_decision_boundary(epoch, logs)
# 训练模型并动态显示
plot_callback = PlotCallback()
model.fit(X_train, y_train, epochs=50, batch_size=16, callbacks=)
# 评估模型
loss, accuracy = model.evaluate(X_test, y_test)
print(f"Test Loss: {loss}")
print(f"Test Accuracy: {accuracy}")
plt.show()
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]