如何保存和加载 NeuralFit model 或权重？

Question

我已经开发了一个 neural.network 来使用neuralfit库学习y=x^2 ，但我想保存 model 以便稍后进行预测。 我目前有：

import neuralfit
import numpy as np

# y(x) = x^2
x = np.arange(10)
y = x**2

# Evolve model
model = neuralfit.Model(1,1)
model.compile('alpha', loss='mse', monitors=['size'])
model.evolve(x,y,epochs=1000)

# Save model
...

我将如何保存和加载model ？

Answer 1

有两种方法可以做到这一点：(1) 使用 Neuralfit 和 (2) 使用 Keras。最好使用 NeuralFit，因为生成的保存文件要小得多（在本例中为 50x）。

方法一：用NeuralFit

根据文档：

# Save a model
model.save('model.nf')

# Load a saved model
model = neuralfit.load('model.nf')

方法二：用Keras

由于 NeuralFit 允许转换为 Keras，我们可以将 model 转换为 Keras，然后使用其功能保存。 换句话说：

# Save a model
keras_model = model.to_keras()
keras_model.save('model.h5')

# Load a saved model
keras_model = keras.models.load_model('model.h5')
model = neuralfit.from_keras(keras_model)