协同过滤模型中的小批量梯度下降

问题描述

电影预测的协同过滤模型可以定义为-

Y = X*Theta'

其中 Y 是电影 * 用户矩阵
X 是一个电影 * n 矩阵
Theta 是一个 (users * n) 矩阵
n 是嵌入维度。

梯度下降可以定义为

Cost = ((X * Theta') - Y)^2  
X_grad = 2 * ((X * Theta') - Y) * Theta  
Theta_grad = 2 * ((X * Theta') - Y) * X

我们在迭代数据集时同时学习 Theta 和 X。

我们可以为协同过滤模型实现小批量梯度下降和/或随机梯度下降吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)