grouped_fe.smooth_embeddings_by_knn

Guide & worked example · Source

grouped_fe.smooth_embeddings_by_knn(
    embeddings,
    neighbor_indices,
    include_self=True,
)

Average each embedding with its nearest neighbors.