Python 如何将keras模型与其他数据一起保存并一起加载?
我有以下代码来训练keras神经网络Python 如何将keras模型与其他数据一起保存并一起加载?,python,arrays,keras,save,load,Python,Arrays,Keras,Save,Load,我有以下代码来训练keras神经网络 from keras import Sequential from keras.layers import Dense from keras.models import load_model import numpy as np class Model: def __init__(self, data=None): self.data = data self.metrics = [] self.mo
from keras import Sequential
from keras.layers import Dense
from keras.models import load_model
import numpy as np
class Model:
def __init__(self, data=None):
self.data = data
self.metrics = []
self.model = self.__build_model()
def __build_model(self):
model = Sequential()
model.add(Dense(4, activation='relu', input_shape=(3,)))
model.add(Dense(1, activation='relu'))
model.compile(loss='mean_squared_error', optimizer='adam', metrics=['accuracy'])
return model
def train(self, epochs):
self.model.fit(self.data[:, :-1], self.data[:,-1], epochs=epochs)
return self
def test(self, data):
self.metrics = self.model.evaluate(data[:, :-1], data[:, -1])
return self
def predict(self, input):
return self.model.predict(input)
def save(self, path):
self.model.save(path)
# I would like to save self.metrics at the same time
def load(self, path):
self.model = load_model(path)
if __name__ == '__main__':
train_data = np.random.rand(1000, 4)
test_data = np.random.rand(100, 4)
print("TRAINING, TESTING & SAVING..")
model = Model(train_data)\
.train(epochs=5)\
.test(test_data)\
.save('./model.h5')
print('LOADING model & PREDICTING..')
test_sample = np.random.rand(1, 3)
model = Model()
model.load('./model.h5')
# I can then do like:
test_output = model.predict(test_sample)
print(test_output)
# And want to get metrics which i had saved with it like:
metrics = model.metrics
print(metrics)
如您所见,它将模型保存到h5文件中,但只保存keras模型,而不保存其他任何内容。
如何在加载keras模型的同时保存其他数据(如指标),然后也能够加载它们
谢谢 您可以使用任何序列化框架来实现这一点
import hickle
def save(self, path):
self.model.save(path)
hkl.dump(self.metrics, 'metrics.hkl', mode='w')
def load(self, path):
self.model = load_model(path)
self.metrics = hkl.load('metrics.hkl')
您还可以将其保存为单个文件,只需从度量和模型对象中创建一个列表或另一个对象。我建议把它们分开保存