oxicuda-recsys 0.2.0

Recommender-system primitives for OxiCUDA — ALS/BPR/NMF, NCF, Two-Tower, DeepFM/AutoInt, SASRec/BERT4Rec, LightGCN/NGCF, MMoE/PLE/ESMM, negative sampling, ranking metrics
Documentation
1
2
3
4
5
6
7
8
9
pub mod autoint;
pub mod cin;
pub mod dcn;
#[allow(clippy::module_inception)]
pub mod deepfm;
pub mod wide_deep;

pub use cin::{CinConfig, CinLayer, XDeepFm};
pub use dcn::{CrossKind, Dcn, DcnConfig};