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
# PinSage model # PinSage model
NOTE: this version is not using NodeFlow yet. 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 First, download and extract from https://dgl.ai.s3.us-east-2.amazonaws.com/dataset/ml-1m.tar.gz
......
...@@ -7,7 +7,6 @@ import tqdm ...@@ -7,7 +7,6 @@ import tqdm
from rec.model.pinsage import PinSage from rec.model.pinsage import PinSage
from rec.datasets.movielens import MovieLens from rec.datasets.movielens import MovieLens
from rec.utils import cuda from rec.utils import cuda
from rec.adabound import AdaBound
from dgl import DGLGraph from dgl import DGLGraph
import argparse 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