Skip to content

GitLab

Unverified Commit bfc2d303 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[BUG] Fix MovieLens feature assignment (#1183) 

Fixes #1182 .

This is embarrassing.
parent bb6a6476
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/recommendation/rec/datasets/movielens.py
View file @ bfc2d303
...@@ -155,7 +155,7 @@ class MovieLens(object): ...@@ -155,7 +155,7 @@ class MovieLens(object):
# bag-of-words # bag-of-words
g.ndata['title'] = torch.zeros(g.number_of_nodes(), len(vocab)) g.ndata['title'] = torch.zeros(g.number_of_nodes(), len(vocab))
for i, tw in enumerate(tqdm.tqdm(title_words)): for i, tw in enumerate(tqdm.tqdm(title_words)):
g.ndata['title'][i, [vocab_invmap[w] for w in tw]] = 1 g.ndata['title'][len(user_ids) + i, [vocab_invmap[w] for w in tw]] = 1
self.vocab = vocab self.vocab = vocab
self.vocab_invmap = vocab_invmap self.vocab_invmap = vocab_invmap
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment