Chemprop 代码相关

20210914

跑实验，存 pkls 来画 auc 图

conda activate chem
git checkout get_fold

第一个实验

python train.py --data_path ~/chemprop/CTSL/CTSL_train_1.csv --dataset_type classification --save_dir checkpoints/CTSL_first_exp --gpu 6 --num_folds 10 --config_path ~/chemprop/CTSL/CTSL.json --ensemble_size 2 &> train_first_exp.log &

第二个实验

python train.py --data_path ~/CTSL_train_2.csv.selfunique.csv  --dataset_type classification --save_dir checkpoints/CTSL_more_forauc --gpu 5 --num_folds 10 --config_path CTSL/CTSL_best.json --ensemble_size 2 &> train_more_auc.log &