-
Notifications
You must be signed in to change notification settings - Fork 3
/
get_eval_results.pbs
28 lines (25 loc) · 910 Bytes
/
get_eval_results.pbs
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
#!/bin/bash
#PBS -N eval
#PBS -e eval.out
#PBS -o eval.out
#PBS -q ai
#PBS -l select=1:ncpus=32:ngpus=1:mem=32gb
#PBS -l walltime=72:00:00
cd $PBS_O_WORKDIR
#module purge
#module load Anaconda3
export PATH=/home/lisong/software/anaconda3/bin:$PATH
export PATH=/home/lisong/local/bin:$PATH
export PATH=/home/lisong/software/openbabel/bin:/home/lisong/software/LeDOCK:$PATH
source activate LS-MolGen
PROTEIN=EGFR
protein=egfr
for model in aae vae latentgan reinvent pocket2mol moldqn sbmolgen lsmolgen; do
for i in {0..5}; do
python calc_metrics.py --gen_path data/${PROTEIN}/${model}/${protein}_${model}_gen_${i}.smi \
--train_path data/chembl31.smi --test_path data/$PROTEIN/${PROTEIN}_pX_7.smi \
--output ./result/${protein}_${model}_gen_${i}_eval_result.csv --n_jobs 32 --device 'cuda' \
--dock_file_dir data/$PROTEIN/ledock \
# --score_path data/baseline/${protein}_${model}_gen.smi
done
done