From fea9649a748844cca1201bf3612e6172f0ccfd19 Mon Sep 17 00:00:00 2001 From: quillotm Date: Thu, 12 Aug 2021 14:30:34 +0200 Subject: [PATCH] Add many measures to compute --- volia/clustering.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/volia/clustering.py b/volia/clustering.py index cf22299..9518fa8 100644 --- a/volia/clustering.py +++ b/volia/clustering.py @@ -8,7 +8,7 @@ import pickle from clustering_modules.kmeans import kmeans from sklearn.preprocessing import LabelEncoder -from sklearn.metrics import v_measure_score +from sklearn.metrics import v_measure_score, homogeneity_score, completeness_score import core.measures import json @@ -20,7 +20,10 @@ CLUSTERING_METHODS = { EVALUATION_METHODS = { "entropy": core.measures.entropy_score, - "v-measure": v_measure_score + "purity": core.measures.purity_score, + "v-measure": v_measure_score, + "homogeneity": homogeneity_score, + "completeness": completeness_score, } -- 1.8.2.3