Merge pull request #224 from mapio/add_add_compound_classes_to_settings

Added add_compound_classes parameter to SettingsTrainingModels
iomega · Nov 23, 2023 · f812065 · f812065
2 parents 0733c60 + cdfde6e
commit f812065
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ms2query/create_new_library/train_models.py b/ms2query/create_new_library/train_models.py
@@ -33,7 +33,7 @@ def __init__(self,
         self.ms2ds_epochs: int = default_settings["ms2ds_epochs"]
         self.ms2query_fraction_for_making_pairs: int = default_settings["ms2query_fraction_for_making_pairs"]
         self.spec2vec_iterations = default_settings["spec2vec_iterations"]
-        self.add_compound_classes = default_settings["add_compound_classes"]
+        self.add_compound_classes: bool = default_settings["add_compound_classes"]
 
 
 def train_all_models(annotated_training_spectra,