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
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use thiserror::Error;

#[derive(Debug, Error)]
pub enum RecsysError {
    #[error("empty input")]
    EmptyInput,
    #[error("dimension mismatch: expected {expected}, got {got}")]
    DimensionMismatch { expected: usize, got: usize },
    #[error("invalid number of users: {n}")]
    InvalidNumUsers { n: usize },
    #[error("invalid number of items: {n}")]
    InvalidNumItems { n: usize },
    #[error("invalid embedding dim: {d}")]
    InvalidEmbeddingDim { d: usize },
    #[error("invalid loss weight: {w}")]
    InvalidLossWeight { w: f32 },
    #[error("invalid k={k} with n={n}")]
    InvalidK { k: usize, n: usize },
    #[error("no negative available for user {user}")]
    NoNegativeAvailable { user: usize },
    #[error("ALS did not converge after {iter} iterations")]
    AlsNotConverged { iter: usize },
    #[error("empty interaction data")]
    EmptyInteraction,
    #[error("unknown user id: {id}")]
    UnknownUser { id: usize },
    #[error("unknown item id: {id}")]
    UnknownItem { id: usize },
    #[error("model not fitted")]
    NotFitted,
    #[error("internal error: {msg}")]
    Internal { msg: String },
    #[error("matrix is not positive definite")]
    NotPositiveDefinite,
    #[error("invalid lambda regularization: {val}")]
    InvalidLambda { val: f32 },
    #[error("item index {idx} out of bounds (n_items={n})")]
    ItemOutOfBounds { idx: usize, n: usize },
    #[error("invalid config: {msg}")]
    InvalidConfig { msg: String },
}

pub type RecsysResult<T> = std::result::Result<T, RecsysError>;