gcn代码pytorch_GCN原理与实现（4）

结合CapsGNN进行图卷积神经网络的学习与pytorch geometric的入门

import torch 
import torch_geometric as tg
from torch_geometric.data import Data
from capsgnn import CapsGNN
import pandas
import numpy
#Then import the CapsGNN source code 
from capsgnn import CapsGNNTrainer
from utils import tab_printer
from param_parser import parameter_parser
#capsgnn,layers,main,param_parser,utils
from param_parser import parameter_parser
args = parameter_parser()
args = args.parse_known_args()[0]

print(args)
'''
THE OUTPUT
Namespace(batch_size=32, capsule_dimensions=8, epochs=100, gcn_filters=20, gcn_layers=2, inner_attention_dimension=20, lambd=0.5, 
learning_rate=0.01, number_of_capsules=8, prediction_path='./output/watts_predictions.csv', test_graph_folder='./input/test/', theta=0.1,
 train_graph_folder='./input/train/', weight_decay=1e-06)
'''