37.68K
Category: programmingprogramming

Сохранение и загрузка моделей Keras

1.

Сохранение и загрузка
моделей Keras

2.

Сохранение модели Keras:
model = ... # Get model (Sequential, Functional Model, or Model subclass)
model.save('path/to/location')
Загрузка модели обратно:
from tensorflow import keras
model = keras.models.load_model('path/to/location')

3.

SavedModel — это более полный формат сохранения, который
сохраняет архитектуру модели, веса и отслеживаемые подграфы
Tensorflow функций вызова.
Это позволяет Keras восстанавливать как встроенные слои, так и
пользовательские объекты.

4.

Использовать модель
y = model.predict(X)
English     Русский Rules