# Dataset features
NAME = 'name'
VOCAB_SIZE = 'vocab_size'
TRUE_CLUSTER_COUNT = 'true_cluster_count'
TOKEN_COUNT_AVG_ENTROPY_A0 = 'token_count_avg_entropy_a0'
TOKEN_COUNT_AVG_ENTROPY_A1 = 'token_count_avg_entropy_a1'
TOKEN_COUNT_SUM_ENTROPY = 'token_count_sum_entropy'
TOKEN_COUNT_FLAT_ENTROPY = 'token_count_flat_entropy'
INTRA_CLUSTER_SPREAD = 'intra_cluster_spread'
INTER_CLUSTER_SPREAD = 'inter_cluster_spread'
SILHOUETTE_SCORE = 'silhouette_score'
CALINSKI_HARABASZ_SCORE = 'calinski_harabasz_score'
DAVIES_BOULDIN_SCORE = 'davies_bouldin_score'

# Mean impurity differences features
PROVIDED_LABELS_COUNT = 'provided_labels_count'
AVG_UNLAB_IMPURITY = 'avg_unlab_impurity'
AVG_LAB_IMPURITY = 'avg_lab_impurity'
VAR_UNLAB_IMPURITY = 'var_unlab_impurity'
VAR_LAB_IMPURITY = 'var_lab_impurity'

# Sample impurity features
PERCENTAGE_LABELED = 'percentage_labeled'
LAB_IMPURITY = 'lab_impurity'
UNLAB_IMPURITY = 'unlab_impurity'
