python - KerasClassifier - Sci-Kit 学习管道 [属性错误]

Question

我在最后使用带有 KerasClassfier 的 Sci-kit 学习管道。分类器将加载经过训练的模型进行预测。但是在将分类器添加到管道后(总共 3 个组件)，我在调用 pipeline.predict_proba 后收到错误 AttributeError: 'KerasClassifier' object has no attribute 'model'。我认为它希望我适合它，但我正在导入训练有素的模型。我在网上找不到任何相关的东西。非常感谢您的帮助。以下是我的代码的相关部分:

def buildEngModelByLoading():
  # load json and create model
  json_file = open('saved_model/cnnModel.json', 'r')
  loaded_model_json = json_file.read()
  json_file.close()
  loaded_model = model_from_json(loaded_model_json)
  # load weights into new model
  loaded_model.load_weights("saved_model/cnnModel.h5")
  print("Loaded classifier model")
  return loaded_model

engSklearnCnn = KerasClassifier(build_fn=buildEngModelByLoading, epochs=20, batch_size=batchSize, verbose=1)
#Append classfier to one pipeline
pipeline.steps.insert(2,['classifier',engSklearnCnn])`

Answer 1

那是因为你忘了在predict函数之前先使用compile方法。

buildEngModelByLoading().compile(optimizer = 'classifier_optimizer', loss = 'loss_function', metrics = 'metrics')

然后只需将 classifier_optimizer、loss_function、metrics 替换为您使用的参数即可。