gru4rec易用性问题
Created by: JiaXiao243
- 负采样使用bpr loss、cross entrop loss的模型结果均保存在 model_bpr_recall20目录下(依次执行会在该目录下覆盖),建议修改为model_neg_recall20。因为使用负采样cross entrop loss训练看到model_bpr_recall20会以为仅存放bpr loss的结果,会不知道负采样cross entrop loss训练出的模型存放在哪里。
- 文档实例训练的模型输出结果建议和infer输入的模型路径一致。 训练示例模型保存路径 infer示例模型加载路径
- 为保证文档前后一致性,训练部分有全词表、负采样得示例,infer部分仅有全词表的示例,建议加负采样预测的示例。 4.预测log中step输出中仅有数据,未标明数据含义,建议添加数据含义recall@20,并优化格式:去掉括号、单引号、空格;或在log中省略该部分。