seqme.utils.knn_alignment_score

seqme.utils.knn_alignment_score#

seqme.utils.knn_alignment_score(xs, labels, n_neighbors=5)[source]#

Compute the k-NN feature alignment score of an embedding model.

Parameters:
  • xs (ndarray) – Sequence embeddings.

  • labels (ndarray) – Group label for each sequence.

  • n_neighbors (int) – Number of neighbors used by k-NN.

Return type:

float

Returns:

Feature alignment score between [0, 1].