我正在尝试使用predict_generator对我的NPL模型进行预测,我的测试数据没有Y变量。我的目的是预测未来文本的标签。我在生成器中尝试过“Y_test=None”,它给了我一个位置错误。我还尝试使用 model.predict(X_test) 进行预测,但这也不起作用。我正在寻求帮助来解决这个问题。请参阅下面的代码,任何建议都是 wolcome。
train_generator = generator (X_train,Y_train,tokenizer,onehot,label_encoder, n_classes,batch_size)
validation_generator = generator (X_valid,Y_valid,tokenizer,onehot,label_encoder, n_classes,batch_size)
test_generator = generator (X_test,Y_test,tokenizer,onehot,label_encoder, n_classes,batch_size)
model.predict_generator(test_generator)
我终于意识到我不需要使用生成器来测试数据,我只需使用分词器来转换 X_train 然后使用 model.predict(X_train) 就可以预测标签了。