Pytorch Embedding Layers RecSys
class RecommendationSystemModel(nn.Module): def __init__( self, num_users, num_movies, embedding_size=128, hidden_dim=128, dropout_rate=0.2, ): super(RecommendationSystemModel, self).__init__() self.num_users = num_users self.num_movies = num_movies self.embedding_size = embedding_size self.hidden_dim = hidden_dim # Embedding layers self.user_embedding = nn.Embedding( num_embeddings=self.num_users, embedding_dim=self.embedding_size ) self.movie_embedding = nn.Embedding( num_embeddings=self.num_movies, embedding_dim=self.embedding_size ) print(self.num_users, self.num_movies) # Hidden layers self.fc1 = nn.Linear(2 * self.embedding_size, self.hidden_dim) self.fc2 = nn.Linear(self.hidden_dim, 1) […]