search

zhaogaofeng611 / TextMatch

基于Pytorch的，中文语义相似度匹配模型（ABCNN、Albert、Bert、BIMPM、DecomposableAttention、DistilBert、ESIM、RE2、Roberta、SiaGRU、XlNet）
787 stars 146 forks source link

如果获取相似的概率，是否取_, out_classes = output_probabilities.max(dim=1)中的_ #13

Open luckyclove opened 1 year ago

luckyclove commented 1 year ago

想获得两个句子相似的概率，_, out_classes = output_probabilities.max(dim=1)的_是否是两个句子相似的概率