search

thu-coai / cotk

Conversational Toolkit. An Open-Source Toolkit for Fast Development and Fair Evaluation of Text Generation
Apache License 2.0
128 stars 27 forks source link

[Feature] 增加SentenceBERT #419

Closed t1101675 closed 4 years ago

t1101675 commented 4 years ago

Describe the feature Step1: 增加Field里面SentenceBERT的支持 Step2: 增加SessionBERT、SentenceCandidateBERT

Expected behavior Step1: LanguageGeneration\SingleTurnDialog\SentenceClassification可以指定pretrain=bert Step2: MultiTurnDialog

Additional context Add any other context about the feature here. 每一个step需要相应测试

hzhwcmhf commented 4 years ago

Done