diff options
Diffstat (limited to 'script/prepare_dataset_test.sh')
| -rw-r--r-- | script/prepare_dataset_test.sh | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/script/prepare_dataset_test.sh b/script/prepare_dataset_test.sh new file mode 100644 index 0000000..6d9db26 --- /dev/null +++ b/script/prepare_dataset_test.sh @@ -0,0 +1,20 @@ +#!/bin/bash +echo "Dataset filename: $1" +echo "Ratio: $2" +echo "Additional arg: $3" +echo "Arg1 for distilling: $4" +echo "Arg2 for distilling: $5" +echo "Arg3 for distilling: $6" +echo "Arg4 for distilling: $7" +echo "Additional arg for merging: $8" + +if [ "$2" != "no_distill" ] +then + #cp ../data/raw_dataset/mawi_ws_ds_sorted.csv.raw ../data/raw_dataset/$1.csv.raw + python3 preprocess_dataset.py --dataset ../data/raw_dataset/$1.csv.raw --kitsune-dataset ../data/raw_dataset/$1.csv.raw.kitsune --distilled-dataset ../data/processed_dataset/$1.csv --ratio $2 --use-direction $4 $5 $6 $7 +fi + +python3 inject_attack.py --dataset ../data/processed_dataset/$1.csv.test --attack-type 1 --use-direction --attack-dataset ../data/processed_dataset/$1.csv.test.attack + + + |