【文件属性】:
文件名称:蔡氏方程matlab求解代码-pytorch-fm:PyTorch中的分解机模型
文件大小:6.26MB
文件格式:ZIP
更新时间:2021-05-19 21:44:53
系统开源
蔡氏方程matlab重构代码PyTorch中的分解机模型
该软件包在CTR预测中提供了分解机模型和常见数据集的PyTorch实现。
可用数据集
可用型号
模型
参考
逻辑回归
分解机
现场感知分解机
高阶分解机
分解支持的神经网络
宽深
注意分解机
神经分解机
神经协同过滤
现场感知神经分解机
产品神经网络
深层交叉网络
深度调频
xDeepFM
AutoInt(自动要素交互模型)
AFN(自适应要素网络模型)
每个模型的AUC值(对于criteo数据集)约为0.80,对于avazu数据集约为0.78。
(请参阅
)
安装
pip
install
torchfm
API文档
执照
麻省理工学院