diff options
| author | Scott <[email protected]> | 2022-01-06 12:13:34 -0800 |
|---|---|---|
| committer | Scott <[email protected]> | 2022-01-06 12:13:34 -0800 |
| commit | 5f75cf834725ac26b289cc5f157af0cb71cd5f0e (patch) | |
| tree | f31baf4247a7d29eb1457a74f75d373d10539237 /ml | |
| parent | ba223f821df1e9b8abbb6f6d23d5ba92c5a70b05 (diff) | |
A bunch of changes...
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) |
