AGCN
修改自AEN,代码中没有修改aen的名字,所以训练时使用参数aen即为AGCN
train_lap.py --model_name aen --dataset restaurant --learning_rate 0.005 --logdir aen_rest_logs
train.py --model_name aen --dataset restaurant --learning_rate 0.005 --logdir aen_rest_logs
Python3.6
所需包如下:
Click 7.0
Keras-Applications 1.0.7 1.0.8
Keras-Preprocessing 1.0.9 1.1.0
Markdown 3.1 3.1.1
Pillow 6.0.0
Werkzeug 0.15.2 0.16.0
absl-py 0.7.1 0.9.0
astor 0.7.1 0.8.1
boto 2.49.0
boto3 1.9.130 1.11.4
botocore 1.12.130
certifi 2019.3.9 2019.11.28
chardet 3.0.4 3.0.4
cycler 0.10.0 0.10.0
docopt 0.6.2 0.6.2
docutils 0.14
gast 0.2.2
gensim 3.7.2
grpcio 1.20.1
h5py 2.9.0
idna 2.8 2.8
jmespath 0.9.4 0.9.4
joblib 0.13.2
kiwisolver 1.1.0 1.1.0
lxml 4.3.3
matplotlib 3.0.3
mock 3.0.3
nltk 3.2.5
numpy 1.16.2
openfile 0.0.7 0.0.7
pip 19.0.3
protobuf 3.7.1
pyparsing 2.4.0
python-dateutil 2.8.0
pytorch-pretrained-bert 0.6.1 0.6.2
regex 2019.3.12
requests 2.21.0 2.22.0
s3transfer 0.2.0 0.3.1
sacremoses 0.0.19 0.0.38
scikit-learn 0.20.3
scipy 1.2.1
setuptools 39.1.0
simplejson 3.16.0 3.17.0
six 1.12.0
sklearn 0.0 0.0
smart-open 1.8.3
tensorboard 1.13.1 2.1.0
tensorboardX 1.6 2.0
tensorflow 1.13.1 2.1.0
tensorflow-estimator 1.13.0 2.1.0
termcolor 1.1.0 1.1.0
toolwrapper 0.3.0 1.0.0
torch 1.0.1
torchsummary 1.5.1 1.5.1
torchsummaryX 1.1.0 1.3.0
torchtext 0.3.1 0.5.0
torchvision 0.2.2.post3 0.1.6
tqdm 4.31.1
urllib3 1.24.1 1.25.7