Views
No views yet
1from transformers import AutoModel
2import torch
3
4# Load model
5model = AutoModel.from_pretrained("matjs/movie_recommendation_tt_small")
6
7# Predict rating
8user_id = torch.tensor([123])
9movie_id = torch.tensor([456])
10genre_features = torch.tensor([[1, 0, 1, 0, 0]]) # One-hot genres
11
12rating = model(user_id, movie_id, genre_features)