Skip to content

Commit a82f9ce

Browse files
committed
add script to run experiements
1 parent ce3ecc3 commit a82f9ce

File tree

3 files changed

+53
-0
lines changed

3 files changed

+53
-0
lines changed

src/gat.sh

+17
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
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

src/gcn.sh

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
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+

src/pscn.sh

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
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+

0 commit comments

Comments
 (0)