[BUG] Fix MovieLens feature assignment (#1183)

Fixes #1182 . This is embarrassing.

[BUG] Fix MovieLens feature assignment (#1183)
Fixes #1182 . This is embarrassing.
bfc2d303 · Quan (Andy) Gan · GitHub · bb6a6476 · bfc2d303
Unverified Commit bfc2d303 authored Jan 09, 2020 by Quan (Andy) Gan Committed by GitHub Jan 09, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/recommendation/rec/datasets/movielens.py examples/pytorch/recommendation/rec/datasets/movielens.py +1 -1

No files found.
--- a/examples/pytorch/recommendation/rec/datasets/movielens.py
+++ b/examples/pytorch/recommendation/rec/datasets/movielens.py
@@ -155,7 +155,7 @@ class MovieLens(object):
        # bag-of-words
        g.ndata['title'] = torch.zeros(g.number_of_nodes(), len(vocab))
        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_invmap = vocab_invmap