首页 > 解决方案 > Predictor.from_archive 失败

问题描述

    archive = load_archive(
    "elmo-constituency-parser-2018.03.14.tar.gz"
)
predictor = Predictor.from_archive(archive, 'constituency-parser')

predictor.predict_json({"sentence": "This is a sentence to be predicted!"})

加载 elmo-constituency-parser 会引发此错误:allennlp.common.checks.ConfigurationError: ptb_trees not in可接受的选择

dataset_reader.type:['babi','conll2003','interleaving','multitask','sequence_tagging','sharded','text_classification_json','multitask_shim']。您应该使用 --include-package 标志来确保加载了正确的模块,或者在配置文件中使用完全限定的类名,例如 {"model": "my_module.models.MyModel"} 自动导入它.

似乎 load_archive 函数返回了一个模型名称“ptb_trees”和一个包含“。”的名称。是必需的,例如 {"model": "my_module.models.MyModel"}

有什么建议吗?谢谢!

标签: allennlp

解决方案


推荐阅读