feat: add save_target_stats method to Artifacts for saving target class statistics

Author: janezlapajne

source/analysis/artifacts.py

@@ -188,5 +188,11 @@ def load_spectral_bands(self) -> Optional[np.ndarray]:
             logger.warning(f"Could not load spectral bands: {e}")
         return None
 
+    def save_target_stats(self, target: np.ndarray):
+        save_path = self._set_save_path(RESULTS)
+        unique, counts = np.unique(target, return_counts=True)
+        target_stats = dict(zip(unique.tolist(), counts.tolist()))
+        write_json(target_stats, save_path / "target_stats.json")
+
 
 artifacts = Artifacts()

source/cli/analysis.py

@@ -88,6 +88,7 @@ def generate_metrics(
 
     metrics_ = metrics.calculate_metrics(model_, encoder, X, y)
     artifacts.save_metrics(metrics_)
+    # artifacts.save_target_stats(y)
 
 
 @app.command()