Кто-нибудь знает, как использовать модуль ранжирования Deeppavlov с собственными ответами? Документация (http://docs.deeppavlov.ai/en/master/components/neural_ranking.html) содержит следующий код (для страхового набора данных):
rank_model = build_model(configs.ranking.ranking_insurance_interact, download=True)
predictor = rank_model.pipe[-1][-1]
candidates = ['auto insurance', 'life insurance', 'home insurance']
predictor.rebuild_responses(candidates)
Но когда я использую этот код для набора данных Ubuntu:
rank_model = build_model(configs.ranking.ranking_ubuntu_v2_bert_sep_interact, download=False)
predictor = rank_model.pipe[-1][-1]
candidates = ['environment variable', 'graphical interface', 'broken terminal']
predictor.rebuilt_responces(candidates)
получается, что метода rebuilt_responces
нет:
AttributeError Traceback (most recent call last)
<ipython-input-9-e013af513ae2> in <module>
4 predictor = rank_model.pipe[-1][-1]
5 candidates = ['environment variable', 'graphical interface', 'broken terminal']
----> 6 predictor.rebuilt_responces(candidates)
AttributeError: 'BertSepRankerPredictor' object has no attribute 'rebuilt_responces'
Есть идеи?
Мне пришлось вручную заменить файлы данными. Решение представлено в выпуске github: https://github.com/deepmipt/DeepPavlov/issues/862