lfm模型训练完成后代价的问题

来源:2-5 LFM模型训练

Whplash

2019-04-29

老师在lfm中实现的是 排序推荐强度完成topk推荐 我的需求是 预测评分 所以我在train_data中的lebel(0/1)替换为了电影本身的评分(0-5)进行训练 按理说训练完成的item_vec和user_vec的向量点积结果是近似于电影评分的。
我使用了50次迭代 记录了每次迭代的cost cost计算为train_data中出现的 |每个用户对每个物品评分-预测评分| 的和 记录结果如下 train_data中总共1w多条预测 但产生了3w多的误差 好像优化过程不是很理想 怎么改进呢图片描述

写回答

1回答

David

2019-04-29

首先确定一下model 是否收敛 就是看一下咱们打印出来的loss 函数是否不再变化。

0
2
Whplash
图中还有十行没截出来 不过趋势也能看出来
2019-04-29
共2条回复

个性化推荐算法实战(可用于毕设) BAT大牛亲授

让你掌握一套完整的,能落地的个性化推荐算法体系。可用于毕设。

840 学习 · 253 问题

查看课程