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
10
11
12
13
14
15
pub mod graphrec;
pub mod hgnn;
pub mod kgat;
#[allow(clippy::module_inception)]
pub mod lightgcn;
#[allow(clippy::module_inception)]
pub mod ngcf;
pub mod pinsage;
pub mod ultragcn;

pub use graphrec::GraphRec;
pub use hgnn::{HeteroEdge, Hgnn, HgnnConfig};
pub use kgat::{Kgat, KgatConfig};
pub use pinsage::{PinSage, PinSageConfig};
pub use ultragcn::{UltraGcn, UltraGcnConfig};