matlab交叉验证代码PyTorch
DGCNN
关于
DGCNN(深图卷积神经网络)的PyTorch实现。
检查更多信息。
要求:python
2.7或python
3.6;
火炬>
=
0.4.0
安装
此实现基于戴汉俊的structure2vec图后端。
在“
lib
/”目录下,键入
make
-j4
编译必要的c
++文件。
之后,在此存储库的根目录下,键入
./run_DGCNN.sh
使用默认设置在数据集MUTAG上运行DGCNN。
或输入
./run_DGCNN.sh
DATANAME
FOLD
在数据集=
DATANAME上运行,使用倍数=
FOLD(1-10,对应于在交叉验证实验中用作测试数据的倍数)。
如果将FOLD设置为0,例如键入“
./run_DGCNN.sh
DD
0”,则它将在DD上运行10倍交叉验证,并报告平均准确度。
或者,键入
./run_DGCNN.sh
DATANAME
1
200
将数据集中的最后200张图用作测试图。
折数1将被忽略。
检查“
run_DGCNN.sh”以获取更多选项。
数据集
默认图形数据集存储在“
data
/
DSN
2024-01-26 18:33:28
35.06MB
系统开源
1