recbole.utils.enum_type

class recbole.utils.enum_type.DataLoaderType(value)

Bases: enum.Enum

Type of DataLoaders.

• ORIGIN: Original DataLoader

• FULL: DataLoader for full-sort evaluation

• NEGSAMPLE: DataLoader for negative sample evaluation

FULL = 2

NEGSAMPLE = 3

ORIGIN = 1

class recbole.utils.enum_type.EvaluatorType(value)

Bases: enum.Enum

Type for evaluation metrics.

• RANKING: Ranking metrics like NDCG, Recall, etc.

• INDIVIDUAL: Individual metrics like AUC, etc.

INDIVIDUAL = 2

RANKING = 1

class recbole.utils.enum_type.FeatureSource(value)

Bases: enum.Enum

Source of features.

• INTERACTION: Features from .inter (other than user_id and item_id).

• USER: Features from .user (other than user_id).

• ITEM: Features from .item (other than item_id).

• USER_ID: user_id feature in inter_feat and user_feat.

• ITEM_ID: item_id feature in inter_feat and item_feat.

• KG: Features from .kg.

• NET: Features from .net.

INTERACTION = 'inter'

ITEM = 'item'

ITEM_ID = 'item_id'

KG = 'kg'

NET = 'net'

USER = 'user'

USER_ID = 'user_id'

class recbole.utils.enum_type.FeatureType(value)

Bases: enum.Enum

Type of features.

• TOKEN: Token features like user_id and item_id.

• FLOAT: Float features like rating and timestamp.

• TOKEN_SEQ: Token sequence features like review.

• FLOAT_SEQ: Float sequence features like pretrained vector.

FLOAT = 'float'

FLOAT_SEQ = 'float_seq'

TOKEN = 'token'

TOKEN_SEQ = 'token_seq'

class recbole.utils.enum_type.InputType(value)

Bases: enum.Enum

Type of Models’ input.

• POINTWISE: Point-wise input, like uid, iid, label.

• PAIRWISE: Pair-wise input, like uid, pos_iid, neg_iid.

LISTWISE = 3

PAIRWISE = 2

POINTWISE = 1

class recbole.utils.enum_type.KGDataLoaderState(value)

Bases: enum.Enum

States for Knowledge-based DataLoader.

• RSKG: Return both knowledge graph information and user-item interaction information.

• RS: Only return the user-item interaction.

• KG: Only return the triplets with negative examples in a knowledge graph.

KG = 3

RS = 2

RSKG = 1

class recbole.utils.enum_type.ModelType(value)

Bases: enum.Enum

Type of models.

• GENERAL: General Recommendation

• SEQUENTIAL: Sequential Recommendation

• CONTEXT: Context-aware Recommendation

• KNOWLEDGE: Knowledge-based Recommendation

CONTEXT = 3

DECISIONTREE = 7

GENERAL = 1

KNOWLEDGE = 4

SEQUENTIAL = 2

SOCIAL = 5

TRADITIONAL = 6