File tree 3 files changed +53
-0
lines changed
3 files changed +53
-0
lines changed Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ if [ -z $1 ];
4
+ then
5
+ echo " input directory not found" ;
6
+ exit ;
7
+ else
8
+ echo " read input from '$1 '" ;
9
+ fi
10
+ file=$1
11
+
12
+ set -x
13
+ CUDA_VISIBLE_DEVICES=0 python train.py --tensorboard-log=exp \
14
+ --model=gat --hidden-units=16,16 \
15
+ --heads=8,8,1 --dim=64 --epochs=500 --lr=0.1 --dropout=0.2 --file-dir=$file \
16
+ --batch=1024 --train-ratio=75 --valid-ratio=12.5 \
17
+ --instance-normalization --use-vertex-feature --class-weight-balanced
Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ if [ -z $1 ];
4
+ then
5
+ echo " input directory not found" ;
6
+ exit ;
7
+ else
8
+ echo " read input from '$1 '" ;
9
+ fi
10
+ file=$1
11
+
12
+ set -x
13
+ CUDA_VISIBLE_DEVICES=0 python train.py --tensorboard-log=exp \
14
+ --model=gcn --hidden-units=128,128 \
15
+ --dim=64 --epochs=500 --lr=0.1 --dropout=0.2 --file-dir=$file \
16
+ --batch=1024 --train-ratio=75 --valid-ratio=12.5 \
17
+ --class-weight-balanced --instance-normalization --use-vertex-feature
18
+
Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ if [ -z $1 ];
4
+ then
5
+ echo " input directory not found" ;
6
+ exit ;
7
+ else
8
+ echo " read input from '$1 '" ;
9
+ fi
10
+ file=$1
11
+
12
+ set -x
13
+ CUDA_VISIBLE_DEVICES=0 python train.py --tensorboard-log=exp \
14
+ --model=pscn --file-dir=$file --epochs=500 --dropout=0.2 \
15
+ --dim=64 --lr=0.1 --hidden-units=16,8 --batch=1024 \
16
+ --train-ratio=75 --valid-ratio=12.5 --class-weight-balanced --instance-normalization \
17
+ --use-vertex-feature --neighbor-size=5 --sequence-size=16
18
+
You can’t perform that action at this time.
0 commit comments