Skip to content

1.1.0

Compare
Choose a tag to compare
@shibing624 shibing624 released this 26 Feb 06:48
· 229 commits to master since this release

重写了CoSENT, SentenceBERT模型的训练和预测代码:

  1. 句子匹配模型训练逻辑继承基类SentenceModel,
  2. 新增train_model, eval_model, 代码结构更清晰,
  3. 预测均使用基类的encode实现。