[FIX] utils

BioSystemsUM · Nov 12, 2024 · 5e02523 · 5e02523
1 parent 8375eb9
commit 5e02523
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deepmol/pipeline_optimization/_utils.py b/src/deepmol/pipeline_optimization/_utils.py
@@ -164,7 +164,7 @@ def preset_deepchem_models(trial, data: Dataset) -> list:
     models = ["gat_model", "gcn_model", "attentive_fp_model", "pagtn_model", "chem_ception_model", "dag_model",
               "graph_conv_model", "smiles_to_vec_model", "text_cnn_model", "weave_model", "dmpnn_model"]
     batch_size = trial.suggest_categorical("batch_size_deepchem", [8, 16, 32, 64, 128, 256, 512])
-    epochs = trial.suggest_int("epochs_deepchem", 100, 1000)
+    epochs = trial.suggest_int("epochs_deepchem", 10, 200)
     deepchem_kwargs = {"epochs": epochs}
     if mode == 'classification' or (len(set(mode)) == 1 and mode[0] == 'classification'):
         # , "multitask_irv_classifier_model",