Merge pull request #453 from MannLabs/fix-max-iterations-param-in-cla…

…ssifier

Add missing max iterations parameter

alphadia/workflow/peptidecentric.py

@@ -99,6 +99,7 @@
 
 def get_classifier_base(
     enable_two_step_classifier: bool = False,
+    two_step_classifier_max_iterations: int = 5,
     enable_nn_hyperparameter_tuning: bool = False,
     fdr_cutoff: float = 0.01,
 ):
@@ -110,6 +111,9 @@ def get_classifier_base(
         If True, uses logistic regression + neural network.
         If False (default), uses only neural network.
 
+    two_step_classifier_max_iterations : int
+        Maximum number of iterations withtin .fit_predict() of the two-step classifier.
+
     enable_nn_hyperparameter_tuning: bool, optional
         If True, uses hyperparameter tuning for the neural network.
         If False (default), uses default hyperparameters for the neural network.
@@ -136,6 +140,7 @@ def get_classifier_base(
             first_classifier=LogisticRegressionClassifier(),
             second_classifier=nn_classifier,
             second_fdr_cutoff=fdr_cutoff,
+            max_iterations=two_step_classifier_max_iterations,
         )
     else:
         return nn_classifier
@@ -179,6 +184,9 @@ def init_fdr_manager(self):
                 enable_two_step_classifier=self.config["fdr"][
                     "enable_two_step_classifier"
                 ],
+                two_step_classifier_max_iterations=self.config["fdr"][
+                    "two_step_classifier_max_iterations"
+                ],
                 enable_nn_hyperparameter_tuning=self.config["fdr"][
                     "enable_nn_hyperparameter_tuning"
                 ],