extract_cluster_assignment function

Extract cluster assignments from model