Spaces:
Runtime error
Runtime error
##### setup | |
#!/bin/bash | |
file_name=$(basename $0) | |
current_path=$(pwd) | |
cd /data/yixin/workspace/unl-graph-usenix | |
source activate /data/yixin/anaconda/unlg | |
datasets=("IMDB-BINARY" "MUTAG" "ENZYMES" "IMDB-MULTI" ) | |
models=( "gcn" "gin" "sage" ) | |
entity="mib-nlp" | |
exp_name="adv-run-v3" | |
batch_size=8 | |
methods=( "clean" "rand" "feat" "grad" "inject" "adv") | |
wd=1e-5 | |
adv_train_budgets=( 0.07 0.09 0.11 ) | |
gen_exp_name="main-results-v2" | |
lr=0.01 | |
es_patience=40 | |
seed_default=0 | |
optimizer="adam" | |
budget=0.05 | |
total_epoch=300 | |
max_steps=5000 | |
seeds=("402") | |
# mkdir $current_path/logs/ if not exist | |
mkdir -p $current_path/logs/ | |
mkdir -p $current_path/logs/$exp_name | |
##### | |
##### loop | |
for adv_train_budget in "${adv_train_budgets[@]}"; do | |
for dataset in "${datasets[@]}"; do | |
for model in "${models[@]}"; do | |
for method in "${methods[@]}"; do | |
##### | |
##### main | |
comb_command="for seed in ${seeds[@]} ; do nohup python eval.py --dataset $dataset --model ${model} --method ${method} --lr $lr --exp_name $exp_name --entity $entity --batch_size $batch_size --seed \$seed --early_stop --num_epochs $total_epoch --wd $wd --device $device --es_patience $es_patience --optimizer $optimizer --max_steps $max_steps --adv_train --adv_train_budget $adv_train_budget --gen_exp_name $gen_exp_name > $current_path/logs/$exp_name/$dataset.$model.$method-\$seed-$RANDOM$RANDOM.log 2>&1 ; done; " | |
eval $comb_command & | |
##### | |
##### | |
done; | |
done; | |
done; | |
done; | |
##### |