neelriyer/creating_weights.py

Created July 12, 2020 23:14

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/neelriyer/3dcd16b6c30501bda9c391f5091f5d20.js"></script>
Save neelriyer/3dcd16b6c30501bda9c391f5091f5d20 to your computer and use it in GitHub Desktop.

Download ZIP

Collab_filtering_adding_weights.py

Raw

creating_weights.py

	class EmbeddingDot(nn.Module):
	def __init__(self):
	super().__init__()
	self.u.weight.data.uniform_(0,0.05)
	self.m.weight.data.uniform_(0,0.05)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment