From 2f71809781c1063120bcdca708545a273e41b509 Mon Sep 17 00:00:00 2001 From: Leonardo Schwarz Date: Fri, 7 Jun 2024 11:35:15 +0200 Subject: [PATCH] more interesting params --- .../workflow/proc/cluster_kmeans.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/depiction_targeted_preproc/workflow/proc/cluster_kmeans.py b/src/depiction_targeted_preproc/workflow/proc/cluster_kmeans.py index 27f5e0b..efd20d9 100644 --- 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)