分解机模型的pytorch实现

  1. 您所在的位置：
  2. Python中文网 >
  3. pypi >
  4. torchfm库 >
  5. 正文

torchfm的Python项目详细描述


pytorch中的分解机模型

该软件包提供了一个pytorch实现的因子分解机模型和常见的数据集在ctr预测。

可用数据集

型号

ModelReference
Logistic Regression
Factorization MachineFactorization Machines
Field-aware Factorization MachineField-aware Factorization Machines for CTR Prediction
Factorization-Supported Neural NetworkDeep Learning over Multi-field Categorical Data - A Case Study on User Response Prediction
Wide&DeepWide & Deep Learning for Recommender Systems
Attentional Factorization MachineAttentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks
Neural Factorization MachineNeural Factorization Machines for Sparse Predictive Analytics
Field-aware Neural Factorization MachineField-aware Neural Factorization Machine for Click-Through Rate Prediction
Product Neural NetworkProduct-based Neural Networks for User Response Prediction
Deep Cross NetworkDeep & Cross Network for Ad Click Predictions
DeepFMDeepFM: A Factorization-Machine based Neural Network for CTR Prediction
xDeepFMxDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems
AutoInt (Automatic Feature Interaction Model)AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networks

对于criteo数据集，每个模型的auc值约为0.80，对于avazu数据集，auc值约为0.78。（请参见example code）

安装

pip install torchfm

API文档

https://rixwew.github.io/pytorch-fm

许可证

麻省理工学院

欢迎加入QQ群-->： 979659372 Python中文网_新手群

推荐PyPI第三方库


导 航 栏

项目 链接

标 签

许可证: BSD许可证（BSD 3条款）

作者信息:: 暂无

维护者

pypi user: rixwew rixwew

最新PyPI项目

最新Python常见问题

热门话题
java绘制两个重叠图像   java如何将图像放入源文件夹并在Eclipse中使用   java性能调优：com。麦克亨格。v2。资源库。Basicresourcepool。waitAvailable（）方法占用应用程序的大部分运行时间   当分数的平行值为某个数字或更高时，java将传递的元素设置为true？   无法在java中使用SimpleFileVisitor删除某些文件。尼奥。文件   java从KType中检索注释   hadoop设置作业。java代码中的属性文件来启动oozie操作   java无响应jbutton请求并发性   java停止当前线程，直到调用的线程完成，然后继续它   JList的java大小调整问题？   通过web浏览器运行java小程序时引发异常   java如何处理selenium webdriver自动化上的背靠背警报   java处理函数只会冻结程序   截击中getparams中的java开关盒   用于显示的java getter   java如何在将布局更改为不可见时修复ClassNotFoundException