Commit 3c07f672ad6b92371b677895a2866763cfef1c37
1 parent
fea9649a74
Exists in
master
Solve an error with the purity score.
Showing 1 changed file with 3 additions and 2 deletions Side-by-side Diff
volia/core/measures.py
| ... | ... | @@ -185,8 +185,9 @@ |
| 185 | 185 | |
| 186 | 186 | |
| 187 | 187 | count_matrix = compute_count_matrix(y_truth, y_hat) |
| 188 | - _, purity_cluster_score = compute_purity_score(count_matrix, 1) | |
| 189 | - _, purity_class_score = cluster_purity = compute_purity_score(count_matrix, 0) | |
| 188 | + | |
| 189 | + purity_cluster_score = compute_purity_score(count_matrix, 1) | |
| 190 | + purity_class_score = compute_purity_score(count_matrix, 0) | |
| 190 | 191 | |
| 191 | 192 | K = np.sqrt(purity_cluster_score * purity_class_score) |
| 192 | 193 |