先搭好环境
作者写的更加优良的GCN 代码:
C:\Users\14634\Desktop\GCN\pytorch_geometric-master\pytorch_geometric-master\benchmark\citation\gcn.py
其中在benchmark 下的 citation下的代码更具有好的特性
parser = argparse.ArgumentParser()
parser.add_argument('--dataset', type=str, required=True)
parser.add_argument('--random_splits', type=bool, default=False)
parser.add_argument('--runs', type=int, default=5)
parser.add_argument('--epochs', type=int, default=200)
parser.add_argument('--lr', type=float, default=0.01)
parser.add_argument('--weight_decay', type=float, default=0.0005)
parser.add_argument('--early_stopping', type=int, default=10)
parser.add_argument('--hidden', type=int, default=16)
parser.add_argument('--dropout', type=float, default=0.5)
parser.add_argument('--normalize_features', type=bool, default=True)
其中对于--dataset后面 required=True , 且没有default ,因此需要在运行的时候:
先进入上一级:
接着:
在运行时,需要提供参数名称以及参数。