我们正在尝试使用allennlp在页面中提供的evaluate_rc-lrc预训练模型https://docs.allennlp.org/models/main/#pre-经过训练的模型使用以下colab代码
!pip install allennlp==1.0.0 allennlp-models==1.0.0
!pip install --pre allennlp-models
!pip install -U nltk
from allennlp.predictors.predictor import Predictor
import allennlp_models
# The instance we want to get LERC score for in a JSON format
input_json = {
'context': 'context string',
'question': 'question string',
'reference': 'reference string',
'candidate': 'candidate string'
}
# evaluate_rc-lerc - A BERT model that scores candidate answers from 0 to 1.
# Loads an AllenNLP Predictor that wraps our trained model
predictor = Predictor.from_path(
archive_path='https://storage.googleapis.com/allennlp-public-models/lerc-2020-11-18.tar.gz',
predictor_name='lerc',
cuda_device=0
)
output_dict = predictor.predict_json(input_json)
print('Predicted LERC Score:', output_dict['pred_score'])
错误:ConfigurationError:lerc不是Model的注册名称。您可能需要使用--include包标志来加载自定义代码。或者,您可以使用完全限定的路径指定您的选择,例如{"模型":"my_module.models.MyModel"},在这种情况下,它们将自动正确导入。
即使在使用了pip的最新软件包后,也要安装allennlp型号==2.2.0
得到相同的错误。"evaluate_rc-lrc"的任何工作代码片段都将非常感激,因为它直接使用保存的存档来避免训练/微调
对不起,LERC模型不是allennlp-models
的一部分。这只是演示的一部分http://demo.allennlp.org.我会看看是否能把文件修好。
LERC型号可在https://github.com/anthonywchen/MOCHA.它可能不像其他型号那样方便使用,但应该仍然很容易。
非常抱歉,发现使用git项目https://github.com/anthonywchen/MOCHA软件包的安装非常具有挑战性,例如allennlp torch依赖项的版本、除requirements.txt之外的其他软件包、手动数据迁移,需要内部代码中的cuda功能,最重要的是相对文件夹路径错误,因此具有存档风格的预训练模型导入将有助于所有类型的用户,以避免处理的技术错误