more interesting params

fgcz · Jun 7, 2024 · 2f71809 · 2f71809
1 parent 68619b3
commit 2f71809
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/depiction_targeted_preproc/workflow/proc/cluster_kmeans.py b/src/depiction_targeted_preproc/workflow/proc/cluster_kmeans.py
@@ -26,8 +26,8 @@ def retain_strongest_signals(data: xarray.DataArray, n_features: int) -> xarray.
 def cluster_kmeans(input_netcdf_path: Annotated[Path, Option()], output_netcdf_path: Annotated[Path, Option()]) -> None:
     image = MultiChannelImage.read_hdf5(input_netcdf_path)
     # TODO make configurable
-    n_clusters = 6
-    n_features = 30
+    n_clusters = 10
+    n_features = 50
 
     reduced_data = retain_strongest_signals(image.data_flat.transpose("i", "c"), n_features)