Update metric.py

SubmissionsIn · Feb 28, 2023 · 921da90 · 921da90
1 parent 81942e7
commit 921da90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metric.py b/metric.py
@@ -1,4 +1,4 @@
-from sklearn.metrics import v_measure_score, adjusted_rand_score, accuracy_score
+from sklearn.metrics import normalized_mutual_info_score, adjusted_rand_score, accuracy_score
 from sklearn.cluster import KMeans
 from scipy.optimize import linear_sum_assignment
 from torch.utils.data import DataLoader
@@ -36,7 +36,7 @@ def purity(y_true, y_pred):
 
 
 def evaluate(label, pred):
-    nmi = v_measure_score(label, pred)
+    nmi = normalized_mutual_info_score(label, pred)
     ari = adjusted_rand_score(label, pred)
     acc = cluster_acc(label, pred)
     pur = purity(label, pred)