Commit 165d4538 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by Minjie Wang
Browse files

[Doc] Update README.md for pytorch PinSage example. (#770)

Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.
parent fd3d54ba
......@@ -2,6 +2,8 @@
NOTE: this version is not using NodeFlow yet.
This example only work with Python 3.6+
First, download and extract from https://dgl.ai.s3.us-east-2.amazonaws.com/dataset/ml-1m.tar.gz
One can then run the following to train PinSage on MovieLens-1M:
......
......@@ -7,7 +7,6 @@ import tqdm
from rec.model.pinsage import PinSage
from rec.datasets.movielens import MovieLens
from rec.utils import cuda
from rec.adabound import AdaBound
from dgl import DGLGraph
import argparse
......
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