diff options
Diffstat (limited to 'ml')
| -rw-r--r-- | ml/model_trainer.py | 4 |
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) |