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
#[allow(clippy::module_inception)]
pub mod als;
#[allow(clippy::module_inception)]
pub mod bpr;
pub mod ease;
pub mod ffm;
pub mod fism;
pub mod ials;
#[allow(clippy::module_inception)]
pub mod nmf;
pub mod slim;
pub mod warp;

pub use ease::{Ease, EaseConfig};
pub use ffm::{Ffm, FfmConfig, FfmEntry};
pub use fism::{Fism, FismConfig};
pub use ials::{Ials, IalsConfig};
pub use slim::{SlimConfig, SlimModel};
pub use warp::{
    WarpConfig, WarpResult, harmonic_number, lambda_rank_weights, ndcg_at_k_from_ranked, warp_loss,
    warp_triple_gradient,
};