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, }