Autokeras多个输入预测失败

时间:2020-10-16 17:12:51

标签: python tensorflow keras auto-keras

我拥有以下自动设置的设置,可以使用TextInputStructuredDataInput来训练模型

model=ak.AutoModel(
inputs=[ak.TextInput(),ak.StructuredDataInput(column_names=["token_ratio","length_ratio","duplicate_tokens","contains_size"])],
outputs=ak.ClassificationHead(num_classes=2,loss='binary_crossentropy'),
    max_trials=1,overwrite=True
)
model.fit(
  [train["title"].to_numpy(),train[["token_ratio","length_ratio","duplicate_tokens","contains_size"]].to_numpy()],
    [train["label"]],epochs=1
)
model.predict([sample["title"].to_numpy(),sample[["token_ratio","length_ratio","duplicate_tokens","contains_size"]].to_numpy()])

该培训工作完美,并且产生了Keras模型。 但是当我调用predict函数时,会发生以下错误:

ValueError: Failed to find data adapter that can handle input: (<class 'tuple'> containing values of types {"<class 'tensorflow.python.data.ops.dataset_ops.MapDataset'>"}), <class 'NoneType'>

建议为什么这是错误的?

0 个答案:

没有答案