Я пытаюсь сохранить свою модель глубокого обучения и быть в состоянии загрузить ее. Когда я пытаюсь загрузить его, я получаю следующую ошибку load_model ("test.mod", True / False, False) У меня есть следующий тестовый код:
импортировать тензор потока как tf из тензорного потока.keras.models import Sequential from tenorflow.keras.layers import Dense, Dropout, LSTM import keras
загрузить набор данных
mnist = tf.keras.datasets.mnist # mnist представляет собой набор данных из 28x28 изображений рукописных цифр и их метки (x_train, y_train), (x_test, y_test) = mnist.load_data () # распаковывает изображения в x_train / x_test и метки в y_train / y_test
нормируют его
x_train = x_train / 255,0 x_test = x_test / 255.0
print (x_train.shape) print (x_train [0] .shape)
модель = Sequential ()
объяснить форму ввода
model.add (LSTM (128, input_shape = (x_train.shape [1:]), активация = 'relu', return_sequence = True))
что выпадает, делают ????
model.add (выпадение (0.2))
model.add (LSTM (128, активация = 'relu')) model.add (выпадение (0.1))
model.add (Dense (32, активация = 'relu')) model.add (выпадение (0,2))
model.add (Dense (10, активация = 'softmax') )
opt = tf.keras.optimizers.Adam (lr = 0,001, распад = 1e-6)
model.compile (loss = 'sparse_categorical_crossentropy', оптимизатор = opt, metrics = [ 'точность'],)
model.fit (x_train, y_train, epochs = 3, validation_data = (x_test, y_test))
model.save ("test.mod")
из keras.models import load_model model = load_model ("test.mod", True / False, False)