fixed issue where noth bathy classified

SlideRuleEarth · Nov 12, 2024 · 8f813a6 · 8f813a6
1 parent 20427c8
commit 8f813a6
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/datasets/bathy/docker/oceaneyes/runner.py b/datasets/bathy/docker/oceaneyes/runner.py
@@ -348,15 +348,15 @@ def runClassifier(classifier, classifier_func, num_processes=6):
     "subaqueous_photons": len(subaqueous_df),
     "bathy_photons": len(bathy_df),
     "bathy_strong_photons": len(bathy_df[bathy_df.spot.isin([1,3,5])]),
-    "bathy_linear_coverage":  bathy_df.index_seg.nunique() * 20.0, # number of unique segments with bathymetry X size of each segment in meters
-    "bathy_mean_depth": bathy_df.depth.mean(),
-    "bathy_min_depth": bathy_df.depth.min(),
-    "bathy_max_depth": bathy_df.depth.max(),
-    "bathy_std_depth": bathy_df.depth.std(),
-    "subaqueous_mean_uncertainty": bathy_df.subaqueous_sigma_tvu.mean().item(),
-    "subaqueous_min_uncertainty": bathy_df.subaqueous_sigma_tvu.min().item(),
-    "subaqueous_max_uncertainty": bathy_df.subaqueous_sigma_tvu.max().item(),
-    "subaqueous_std_uncertainty": bathy_df.subaqueous_sigma_tvu.std().item()
+    "bathy_linear_coverage":  float(bathy_df.index_seg.nunique() * 20.0), # number of unique segments with bathymetry X size of each segment in meters
+    "bathy_mean_depth": float(bathy_df.depth.mean()),
+    "bathy_min_depth": float(bathy_df.depth.min()),
+    "bathy_max_depth": float(bathy_df.depth.max()),
+    "bathy_std_depth": float(bathy_df.depth.std()),
+    "subaqueous_mean_uncertainty": float(bathy_df.subaqueous_sigma_tvu.mean()),
+    "subaqueous_min_uncertainty": float(bathy_df.subaqueous_sigma_tvu.min()),
+    "subaqueous_max_uncertainty": float(bathy_df.subaqueous_sigma_tvu.max()),
+    "subaqueous_std_uncertainty": float(bathy_df.subaqueous_sigma_tvu.std())
 }
 
 # read versions