collie
latest
API Reference
Interactions
Cross-Validation
Losses
Models
Evaluation Metrics
Utilities
Tutorials, Examples, and More
MovieLens 100K
Tutorials
Guidelines
Contribution Guidelines
Code of Conduct
collie
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
A
adaptive_bpr_loss() (in module collie.loss)
adaptive_hinge_loss() (in module collie.loss)
advance_stage() (collie.model.MultiStagePipeline method)
ApproximateNegativeSamplingInteractionsDataLoader (class in collie.interactions)
Args (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader attribute)
(collie.interactions.InteractionsDataLoader attribute)
auc() (in module collie.metrics)
B
BasePipeline (class in collie.model)
bpr_loss() (in module collie.loss)
C
calculate_loss() (collie.model.BasePipeline method)
checkpoint_callback (collie.model.CollieTrainer property)
checkpoint_callbacks (collie.model.CollieTrainer property)
ckpt_path (collie.model.CollieTrainer property)
ColdStartModel (class in collie.model)
CollaborativeMetricLearningModel (class in collie.model)
CollieMinimalTrainer (class in collie.model)
CollieTrainer (class in collie.model)
configure_optimizers() (collie.model.BasePipeline method)
(collie.model.MultiStagePipeline method)
convert_to_implicit (class in collie.utils)
create_ratings_matrix (class in collie.utils)
current_epoch (collie.model.CollieTrainer property)
D
DeepFM (class in collie.model)
default_root_dir (collie.model.CollieTrainer property)
device_ids (collie.model.CollieTrainer property)
df_to_html (class in collie.utils)
df_to_interactions (class in collie.utils)
E
early_stopping_callback (collie.model.CollieTrainer property)
early_stopping_callbacks (collie.model.CollieTrainer property)
enable_validation (collie.model.CollieTrainer property)
estimated_stepping_batches (collie.model.CollieTrainer property)
evaluate_in_batches() (in module collie.metrics)
explicit_evaluate_in_batches() (in module collie.metrics)
ExplicitInteractions (class in collie.interactions)
F
fit() (collie.model.CollieMinimalTrainer method)
(collie.model.CollieTrainer method)
forward() (collie.model.BasePipeline method)
(collie.model.ColdStartModel method)
(collie.model.CollaborativeMetricLearningModel method)
(collie.model.DeepFM method)
(collie.model.HybridModel method)
(collie.model.HybridPretrainedModel method)
(collie.model.MatrixFactorizationModel method)
(collie.model.MLPMatrixFactorizationModel method)
(collie.model.NeuralCollaborativeFiltering method)
(collie.model.NonlinearMatrixFactorizationModel method)
freeze_embeddings() (collie.model.HybridPretrainedModel method)
G
get_item_predictions() (collie.model.BasePipeline method)
get_movielens_metadata() (in module collie.movielens)
get_recommendation_visualizations() (in module collie.movielens)
get_user_metadata() (in module collie.movielens)
get_user_predictions() (collie.model.BasePipeline method)
global_step (collie.model.CollieTrainer property)
H
HDF5Interactions (class in collie.interactions)
HDF5InteractionsDataLoader (class in collie.interactions)
head() (collie.interactions.ExplicitInteractions method)
(collie.interactions.HDF5Interactions method)
(collie.interactions.Interactions method)
hinge_loss() (in module collie.loss)
HybridModel (class in collie.model)
HybridPretrainedModel (class in collie.model)
I
Interactions (class in collie.interactions)
interactions (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader attribute)
(collie.interactions.InteractionsDataLoader attribute)
InteractionsDataLoader (class in collie.interactions)
is_last_batch (collie.model.CollieTrainer property)
item_bucket_item_similarity() (collie.model.ColdStartModel method)
item_item_similarity() (collie.model.BasePipeline method)
L
load_from_hybrid_model() (collie.model.HybridPretrainedModel method)
M
mapk() (in module collie.metrics)
mat (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader property)
(collie.interactions.HDF5InteractionsDataLoader property)
(collie.interactions.InteractionsDataLoader property)
MatrixFactorizationModel (class in collie.model)
max_epochs (collie.model.CollieMinimalTrainer property)
(collie.model.CollieTrainer property)
MLPMatrixFactorizationModel (class in collie.model)
model (collie.model.CollieTrainer property)
mrr() (in module collie.metrics)
MultiStagePipeline (class in collie.model)
N
NeuralCollaborativeFiltering (class in collie.model)
NonlinearMatrixFactorizationModel (class in collie.model)
num_devices (collie.model.CollieTrainer property)
num_interactions (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader property)
(collie.interactions.HDF5InteractionsDataLoader property)
(collie.interactions.InteractionsDataLoader property)
num_items (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader property)
(collie.interactions.HDF5InteractionsDataLoader property)
(collie.interactions.InteractionsDataLoader property)
num_negative_samples (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader property)
(collie.interactions.ExplicitInteractions property)
(collie.interactions.HDF5InteractionsDataLoader property)
(collie.interactions.InteractionsDataLoader property)
num_users (collie.interactions.ApproximateNegativeSamplingInteractionsDataLoader property)
(collie.interactions.HDF5InteractionsDataLoader property)
(collie.interactions.InteractionsDataLoader property)
O
on_fit_start() (collie.model.BasePipeline method)
optimizer_step() (collie.model.MultiStagePipeline method)
P
pandas_df_to_hdf5 (class in collie.utils)
predict() (collie.model.CollieTrainer method)
prediction_writer_callbacks (collie.model.CollieTrainer property)
progress_bar_callback (collie.model.CollieTrainer property)
R
random_split() (in module collie.cross_validation)
read_movielens_df() (in module collie.movielens)
read_movielens_df_item() (in module collie.movielens)
read_movielens_posters_df() (in module collie.movielens)
remove_users_with_fewer_than_n_interactions (class in collie.utils)
reset_parameters() (collie.model.ScaledEmbedding method)
(collie.model.ZeroEmbedding method)
reset_predict_dataloader() (collie.model.CollieTrainer method)
reset_test_dataloader() (collie.model.CollieTrainer method)
reset_train_dataloader() (collie.model.CollieTrainer method)
reset_val_dataloader() (collie.model.CollieTrainer method)
run_movielens_example() (in module collie.movielens)
S
save_checkpoint() (collie.model.CollieTrainer method)
save_model() (collie.model.BasePipeline method)
(collie.model.HybridModel method)
(collie.model.HybridPretrainedModel method)
ScaledEmbedding (class in collie.model)
set_stage() (collie.model.ColdStartModel method)
(collie.model.MultiStagePipeline method)
stratified_split() (in module collie.cross_validation)
T
tail() (collie.interactions.ExplicitInteractions method)
(collie.interactions.HDF5Interactions method)
(collie.interactions.Interactions method)
test() (collie.model.CollieTrainer method)
time_since_start() (collie.utils.Timer method)
timecheck() (collie.utils.Timer method)
Timer (class in collie.utils)
toarray() (collie.interactions.ExplicitInteractions method)
(collie.interactions.Interactions method)
todense() (collie.interactions.ExplicitInteractions method)
(collie.interactions.Interactions method)
train_dataloader() (collie.model.BasePipeline method)
training_epoch_end() (collie.model.BasePipeline method)
training_step() (collie.model.BasePipeline method)
trunc_normal (class in collie.utils)
tune() (collie.model.CollieTrainer method)
U
unfreeze_embeddings() (collie.model.HybridPretrainedModel method)
user_user_similarity() (collie.model.BasePipeline method)
V
val_dataloader() (collie.model.BasePipeline method)
validate() (collie.model.CollieTrainer method)
validation_epoch_end() (collie.model.BasePipeline method)
validation_step() (collie.model.BasePipeline method)
W
warp_loss() (in module collie.loss)
Z
ZeroEmbedding (class in collie.model)
Read the Docs
v: latest
Versions
latest
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds