✨ why does training fail (silently?)

wfondrie · Oct 8, 2024 · 78548ff · 78548ff
1 parent f11693e
commit 78548ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/mokapot/brew.py b/mokapot/brew.py
@@ -149,6 +149,7 @@ def brew(
             )
 
     except TypeError:
+        LOGGER.info("No idea how we ended up here, but this seems to be standard training oO?! tf.")
         train_sets = list(
             make_train_sets(
                 test_idx=test_folds_idx,
@@ -539,6 +540,8 @@ def _fit_model(train_set, psms, model, fold):
     try:
         model.fit(train_set)
     except RuntimeError as msg:
+        LOGGER.info("=== Analyzing Fold %i ===", fold + 1)
+        LOGGER.info(msg)
         if str(msg) != "Model performs worse after training.":
             raise
 

diff --git a/mokapot/mokapot.py b/mokapot/mokapot.py
@@ -120,10 +120,6 @@ def main(main_args=None):
         rng=config.seed,
     )
     logging.info("")
-    logging.info("BREW DELETE ME")
-    for s in scores:
-        logging.info(f"{s}")
-        logging.info(f"{(s == 0).all()}")
 
     if config.dest_dir is not None:
         config.dest_dir.mkdir(exist_ok=True)