Skip to content

How to make Tensorrec have stable results #154

@melaniab

Description

@melaniab

Hello,

I am currently using TensorRec for my master's thesis project, and have been following the MovieLens guide on getting started with the library. I am trying to test different features (embeddings) for content-based recommendations. However, the results that I have so far are not stable, meaning I get different results when I rerun the same piece of code with the same data.

For example, with the same dataset I got first

Recall at 10: Train: 0.0424 Test: 0.0407

and then later:

Recall at 10: Train: 0.0412 Test: 0.0406

How can I make it stable? I did not see anywhere in the code random_seed being set, and I'm not sure where and how can I set it.

The RecSys I use is:

content_model = tensorrec.TensorRec(
    n_components=n_features,
    item_repr_graph=tensorrec.representation_graphs.FeaturePassThroughRepresentationGraph(),
     loss_graph=tensorrec.loss_graphs.WMRBLossGraph()
)
content_model.fit(interactions=sparse_train_ratings_4plus,
                  user_features=user_indicator_features,
                  item_features=books_features,
                 n_sampled_items=int(n_items * .01))

Thank you in advance!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions