Классификация интентов в чатботе - сравнение моделей ELMO, BERT и char-level linear sklearn
Сравнение выполнялось по f1 weighted метрике на датасете с ~5100 сэмплов через 8-фолдовую кросс-валидацию. Чуть больше 70 интентов с большим дисбалансом.
Финальные результаты:
LinearSVC ~0.77
BERT ~0.815
ELMo ~0.817
( Пояснения по моделям )
Финальные результаты:
LinearSVC ~0.77
BERT ~0.815
ELMo ~0.817
( Пояснения по моделям )
Comments 1
Reply
Leave a comment