X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=ml%2Fmodel_trainer.py;fp=ml%2Fmodel_trainer.py;h=acd721868a2a9e04de0da364b8d37dcc268b4fee;hb=5f75cf834725ac26b289cc5f157af0cb71cd5f0e;hp=f61b8e745b6cd6c02f23e258003928caba81916b;hpb=ba223f821df1e9b8abbb6f6d23d5ba92c5a70b05;p=python_utils.git

diff --git a/ml/model_trainer.py b/ml/model_trainer.py
index f61b8e7..acd7218 100644
--- a/ml/model_trainer.py
+++ b/ml/model_trainer.py
@@ -246,12 +246,12 @@ class TrainingBlueprint(ABC):
                     y.pop()
 
                 if self.spec.delete_bad_inputs:
-                    msg = f"WARNING: {filename}: missing features or label.  DELETING."
+                    msg = f"WARNING: {filename}: missing features or label; expected {self.spec.feature_count} but saw {len(x)}.  DELETING."
                     print(msg, file=sys.stderr)
                     logger.warning(msg)
                     os.remove(filename)
                 else:
-                    msg = f"WARNING: {filename}: missing features or label.  Skipped."
+                    msg = f"WARNING: {filename}: missing features or label; expected {self.spec.feature_count} but saw {len(x)}.  Skipping."
                     print(msg, file=sys.stderr)
                     logger.warning(msg)
         return (X, y)