def cluster_nmi(Y_pred, Y): assert Y_pred.size == Y.size nmi = normalized_mutual_info_score(Y, Y_pred) return nmi