我正在尝试在python中序列化分类模型。该模型本身是通过一个函数创建的,此后,将创建一个管道对象以完成这些步骤(请参见代码)。
我已经尝试了泡菜和莳萝,但是我的调试器显示它不是有效的训练模型二进制。
def create_model():
# create model
model = Sequential()
model.add(Dense(30, input_dim=77, kernel_initializer='normal', activation='relu'))
model.add(Dense(1, kernel_initializer='normal', activation='sigmoid'))
# Compile model
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
return model
estimators = []
estimators.append(('standardise', StandardScaler()))
estimators.append(('mlp', KerasClassifier(build_fn=create_model, epochs=50, batch_size=5, verbose=0)))
pipeline = Pipeline(estimators)
pickle.dumps(pipeline, file)
文件未保存,而是显示以下错误:
TypeError: an integer is required (got type _io.BufferedWriter)