fastNLP.core.predictor module

undocumented

class fastNLP.core.predictor.Predictor(network)[源代码]

基类:object

一个根据训练模型预测输出的预测器(Predictor)

与测试器(Tester)不同的是,predictor不关心模型性能的评价指标,只做inference。 这是一个fastNLP调用的高级模型包装器。它与Trainer、Tester不共享任何操作。

__init__(network)[源代码]
参数

network (torch.nn.Module) – 用来完成预测任务的模型

predict(data: fastNLP.core.dataset.DataSet, seq_len_field_name=None)[源代码]

用已经训练好的模型进行inference.

参数
  • data (fastNLP.DataSet) – 待预测的数据集

  • seq_len_field_name (str) – 表示序列长度信息的field名字

返回

dict dict里面的内容为模型预测的结果