프로젝트
GitHub에서 보기
rixwew/pytorch-fm
rixwew/pytorch-fm has 1,080 GitHub stars and 227 forks. GitHot measured 0 stars in 24 hours and 0 in 7 days. Last checked 2026-11-15 13:32 UTC. Last push 2024-04-08.
측정된 스타 증가
- in 24h
- 0
- in 7 days
- 0
- in 15 days
- 기록 수집 중
- in 1 month
- 기록 수집 중
서로 겹치는 기간의 순증가량이며 일별 합계가 아닙니다. 측정값이 없으면 비워 둡니다. 스타는 관심도를 나타내며 품질을 보장하지 않습니다.
소개
Factorization Machine models in PyTorch
머신러닝README.md
OriginalFactorization Machine models in PyTorch
This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.
Available Datasets
Available Models
Each model's AUC values are about 0.80 for criteo dataset, and about 0.78 for avazu dataset. (please see example code)
Installation
pip install torchfm
API Documentation
https://rixwew.github.io/pytorch-fm
Licence
MIT