ModelCheckpoint自动保存模型
-- coding: utf-8 --
import tensorflow as tf import json import tensorflow.keras.models from tensorflow.keras.callbacks import * #这是独立的包 import tensorflow.keras mnist = tf.keras.datasets.mnist (x_train, y_train), (x_test, y_test) = mnist.load_data() x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(), tf.keras.layers.Dense(128, activation=‘relu’),
版权声明:
作者:zhangchen
链接:https://www.techfm.club/p/20776.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。
THE END
二维码
共有 0 条评论