InsaneLife / dssm

DSSM and Multi-View DSSM
660 stars 230 forks source link

i want to predict a query pair like "今天天气怎么样" “今天温度怎么样” ,what to do? #25

Closed yangyan1227 closed 4 years ago

JNero commented 5 years ago

+1 。怎么做预测?

InsaneLife commented 5 years ago

分别作为输入,获取输出句子的embedding向量,让后计算两个向量cos相似度即可。

liuyangzy commented 4 years ago

分别作为输入,获取输出句子的embedding向量,让后计算两个向量cos相似度即可。

如何输入

InsaneLife commented 4 years ago

需要先训练模型,然后做预测,训练入口:train.py 训练(默认使用功LCQMC数据集):

python train.py --mode=train

预测:

python train.py --mode=train --file=$predict_file$

测试文件格式: q1\tq2, 例如:

今天天气怎么样 今天温度怎么样
InsaneLife commented 4 years ago

需要先训练模型,然后做预测,训练入口:train.py 训练(默认使用功LCQMC数据集):

python train.py --mode=train

预测:

python train.py --mode=train --file=$predict_file$

测试文件格式: q1\tq2, 例如:

今天天气怎么样   今天温度怎么样

29 #26