-
Notifications
You must be signed in to change notification settings - Fork 221
Open
Description
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!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels