diff options
Diffstat (limited to 'ml')
| -rw-r--r-- | ml/model_trainer.py | 5 | ||||
| -rw-r--r-- | ml/quick_label.py | 7 |
2 files changed, 7 insertions, 5 deletions
diff --git a/ml/model_trainer.py b/ml/model_trainer.py index acd7218..79ce706 100644 --- a/ml/model_trainer.py +++ b/ml/model_trainer.py @@ -12,6 +12,7 @@ import random import sys from types import SimpleNamespace from typing import Any, List, NamedTuple, Optional, Set, Tuple +import warnings import numpy as np from sklearn.model_selection import train_test_split # type:ignore @@ -247,13 +248,13 @@ class TrainingBlueprint(ABC): if self.spec.delete_bad_inputs: 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) + warnings.warn(msg) os.remove(filename) else: 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) + warnings.warn(msg) return (X, y) def make_progress_graph(self) -> None: diff --git a/ml/quick_label.py b/ml/quick_label.py index 1ed4296..120ff5f 100644 --- a/ml/quick_label.py +++ b/ml/quick_label.py @@ -4,6 +4,7 @@ import glob import logging import os from typing import Callable, List, NamedTuple, Optional, Set +import warnings import argparse_utils import config @@ -90,9 +91,9 @@ def label(in_spec: InputSpec) -> None: continue features = in_spec.image_file_to_features_file(image) if features is None or not os.path.exists(features): - logger.warning( - f'File {image} yielded file {features} which does not exist, SKIPPING.' - ) + msg = f'File {image} yielded file {features} which does not exist, SKIPPING.' + logger.warning(msg) + warnings.warn(msg) continue # Render features and image. |