tirank.Imageprocessing.process_embeddings

tirank.Imageprocessing.process_embeddings(embeddings, n_components, n_clusters)

Performs PCA dimensionality reduction and K-means clustering.

Parameters:

• embeddings (torch.Tensor or np.ndarray) – The feature embeddings.

• n_components (int) – The number of principal components to keep.

• n_clusters (int) – The number of clusters to find (k).

Returns:

A tuple containing:

• np.ndarray: The PCA-transformed embeddings.

• np.ndarray: The cluster labels for each spot.

Return type:

tuple