Использование конвейера hugsface fill-mask для получения «балла» за результат, который он не предлагал - PullRequest
0 голосов
/ 12 апреля 2020

Я использовал обнимающее лицо, чтобы делать прогнозы для маскированных токенов, и это прекрасно работает. Я заметил, что для каждого прогноза он дает «оценку» и хотел бы получить «оценку» для некоторых токенов, которые он не предсказывал, но которые я предоставляю.

Например, если мое предложение "I ate bacon and <mask> for breakfast", я могу использовать pipeline('fill-mask'), чтобы получить предсказания и их оценки, например, это может вернуть мне ["eggs", 0.1]. Но то, что я хотел бы сделать, это предоставить свою собственную догадку, а затем вернуть счет, который он присваивает моей собственной догадке. например, я мог бы хотеть узнать, какую оценку он дает слову «блины» в данной ситуации.

Возможно ли это?

1 Ответ

0 голосов
/ 10 мая 2020

Вы можете использовать bert-as-service для этой задачи. См. Эту библиотеку .

. Вы также можете использовать метод fill_mask (). См. Эту ссылку

.
...