models

Recommendation models (rectools.models)

Convenient wrappers for popular recommendation algorithms (ItemKNN, ALS, LightFM), also some custom implementations.

Models

models.DSSMModel models.EASEModel models.ImplicitALSWrapperModel models.ImplicitBPRWrapperModel models.ImplicitItemKNNWrapperModel models.LightFMWrapperModel models.PopularModel models.PopularInCategoryModel models.PureSVDModel models.RandomModel models.nn.bert4rec.BERT4RecModel models.nn.sasrec.SASRecModel

Modules

rectools.models.base

Base model.

rectools.models.ease

EASE model.

rectools.models.implicit_als

rectools.models.implicit_bpr

rectools.models.implicit_knn

rectools.models.lightfm

rectools.models.nn

Recommendation models based on neural nets.

rectools.models.popular

Popular model.

rectools.models.popular_in_category

Popular in category model.

rectools.models.pure_svd

SVD Model.

rectools.models.random

Random Model.

rectools.models.rank

Recommendation models (rectools.models.rank) :2: (WARNING/2) Title underline too short. Recommendation models (rectools.models.rank) ==============================================

rectools.models.ranking

Two-stage ranking Recommendation models (rectools.models.ranking) :2: (WARNING/2) Title underline too short. Two-stage ranking Recommendation models (rectools.models.ranking) ==============================================

rectools.models.serialization

rectools.models.utils

Useful functions.

rectools.models.vector

Base classes for vector models.