Skip to content

Instantly share code, notes, and snippets.

@MathiasGruber
Last active June 23, 2021 09:02
Show Gist options
  • Save MathiasGruber/073099003cd8a348d81f04960bb94b0c to your computer and use it in GitHub Desktop.
Save MathiasGruber/073099003cd8a348d81f04960bb94b0c to your computer and use it in GitHub Desktop.
Sample training of chemprop model
chemprop_train \
--data_path data/clintox_train.csv \
--config_path data/config.json \
--dataset_type classification \
--save_dir model_checkpoint \
--num_folds 5 \
--ensemble_size 3 \
--features_generator rdkit_2d_normalized --no_features_scaling \
--split_type scaffold_balanced
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment