我使用了Bert模型进行了训练并保存为hdf5文件，但在尝试预测时，出现了以下错误：

IndexError: 列表索引超出范围

以下是代码

我试图预测测试集中的一句话。

完整代码错误的图片

感谢您的帮助，谢谢

编辑：

回答：

如ktrain教程和示例笔记本中所示，例如这个，您需要使用Predictor实例来对原始文本输入进行预测：

# 创建Predictor实例predictor = ktrain.get_predictor(learner.model, preproc)# 进行预测output = predictor.predict('我爱这部电影！')print(output)# 将Predictor保存到磁盘predictor.save('/tmp/mypredictor')# 从磁盘重新加载Predictorreloaded_predictor = ktrain.load_predictor('/tmp/mypredictor')# 进行另一个预测output = reloaded_predictor.predict('我爱这部电影！')print(output)