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 |