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)