使用训练有素的BERT模型进行预测

Question

我使用train.tsv和dev.tsv数据集训练了this repo中的BertForSequenceClassification模型。现在，我需要使用训练有素的模型对未标记的数据集进行预测。你们知道如何使用bert模型进行预测吗？我正在寻找类似于Keras的prepare（）函数，以获取未标记的数据作为输入并预测标签。