如何在Python 3中序列化管道对象/函数?

时间:2019-04-29 14:25:59

标签: python serialization pickle pipeline dill

我正在尝试在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)

0 个答案:

没有答案