Spaces:
Paused
Paused
File size: 1,345 Bytes
181b6d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | #!/bin/bash
chemprop_dir=../../../chemprop # location of chemprop directory, CHANGE ME
save_dir=results_timing
data_dir=../data/timing
#100k
python $chemprop_dir/train.py \
--dataset_type regression \
--data_path $data_dir/qm9_100k.csv \
--save_dir $save_dir/qm9_100k \
--aggregation norm \
--depth 4 \
--ffn_num_layers 2 \
--hidden_size 1000 \
--ffn_hidden_size 1000 \
--epochs 50
python $chemprop_dir/predict.py \
--test_path $data_dir/qm9_100k.csv \
--preds_path $save_dir/qm9_100k/preds/preds.csv \
--checkpoint_dir $save_dir/qm9_100k
#10k
python $chemprop_dir/train.py \
--dataset_type regression \
--data_path $data_dir/qm9_10k.csv \
--save_dir $save_dir/qm9_10k \
--aggregation norm \
--depth 4 \
--ffn_num_layers 2 \
--hidden_size 1000 \
--ffn_hidden_size 1000 \
--epochs 50
python $chemprop_dir/predict.py \
--test_path $data_dir/qm9_10k.csv \
--preds_path $save_dir/qm9_10k/preds/preds.csv \
--checkpoint_dir $save_dir/qm9_10k
#1k
python $chemprop_dir/train.py \
--dataset_type regression \
--data_path $data_dir/qm9_1k.csv \
--save_dir $save_dir/qm9_1k \
--aggregation norm \
--depth 4 \
--ffn_num_layers 2 \
--hidden_size 1000 \
--ffn_hidden_size 1000 \
--epochs 50
python $chemprop_dir/predict.py \
--test_path $data_dir/qm9_1k.csv \
--preds_path $save_dir/qm9_1k/preds/preds.csv \
--checkpoint_dir $save_dir/qm9_1k
|