Skip to content

Instantly share code, notes, and snippets.

@neelriyer

neelriyer/collab_filtering_create_optimiser.py

Created July 12, 2020 23:17
Show Gist options
  • Save neelriyer/03dec8d2aa9332ec513438cb3ae89d8b to your computer and use it in GitHub Desktop.
Save neelriyer/03dec8d2aa9332ec513438cb3ae89d8b to your computer and use it in GitHub Desktop.
Download ZIP
Collab filtering create the optimser
Raw
collab_filtering_create_optimiser.py
model = EmbeddingDot(n_users, n_movies).cuda()
opt = optim.SGD(model.parameters(), 1e-1, weight_decay=1e-5, momentum=0.9)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment