【Python】过拟合问题

最新推荐文章于 2023-08-08 07:10:13 发布

Nirvana；

最新推荐文章于 2023-08-08 07:10:13 发布

阅读量267

点赞数 3

分类专栏：机器学习文章标签： python tensorflow 深度学习

本文链接：https://blog.csdn.net/weixin_45355387/article/details/122241948

版权

机器学习专栏收录该内容

7 篇文章 2 订阅

订阅专栏

Tensorflow实现过拟合问题

import tensorflow as tf
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

# 过拟合问题
(train_image, train_label), (test_image, test_label) = tf.keras.datasets.fashion_mnist.load_data()
train_label_onehot = tf.keras.utils.to_categorical(train_label)

test_label_onehot = tf.keras.utils.to_categorical(test_label)

model = tf.keras.Sequential()
model.add(tf.keras.layers.Flatten(input_shape=(28, 28)))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(10, activation='softmax'))

model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.001), loss='categorical_crossentropy', metrics=['acc'])
history = model.fit(train_image, train_label_onehot, epochs=10, validation_data=(test_image, test_label_onehot))

plt.plot(history.epoch, history.history.get('loss'), label='loss')
plt.plot(history.epoch, history.history.get('val_loss'), label='val_loss')
plt.legend()
plt.show()

plt.plot(history.epoch, history.history.get('acc'), label='acc')
plt.plot(history.epoch, history.history.get('val_acc'), label='val_acc')
plt.legend()
plt.show()