summaryrefslogtreecommitdiff
path: root/ml
diff options
context:
space:
mode:
authorScott <[email protected]>2022-01-06 12:13:34 -0800
committerScott <[email protected]>2022-01-06 12:13:34 -0800
commit5f75cf834725ac26b289cc5f157af0cb71cd5f0e (patch)
treef31baf4247a7d29eb1457a74f75d373d10539237 /ml
parentba223f821df1e9b8abbb6f6d23d5ba92c5a70b05 (diff)
A bunch of changes...
Diffstat (limited to 'ml')
-rw-r--r--ml/model_trainer.py4
1 files changed, 2 insertions, 2 deletions
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)