get_partition function

clusters memberships for a partition into K clusters.