diff --git a/src/cev/_compare_metric_dropdown.py b/src/cev/_compare_metric_dropdown.py index dac2fcb..612e53e 100644 --- a/src/cev/_compare_metric_dropdown.py +++ b/src/cev/_compare_metric_dropdown.py @@ -183,7 +183,7 @@ def callback(): "viridis", "viridis_r", value_range_slider.value, - ("Low", "High", "Neighborhood"), + ("Similar", "Dissimilar", "Neighborhood"), ) else: raise ValueError( diff --git a/src/cev/metrics.py b/src/cev/metrics.py index 60a6936..186ac96 100644 --- a/src/cev/metrics.py +++ b/src/cev/metrics.py @@ -36,7 +36,7 @@ def confusion(df: pd.DataFrame) -> pd.Series: def neighborhood(df: pd.DataFrame, max_depth: int = 1) -> pd.DataFrame: categories = df["label"].cat.categories neighborhood_scores = cev_metrics.neighborhood(df, max_depth) - np.fill_diagonal(neighborhood_scores, 1) + np.fill_diagonal(neighborhood_scores, 0) return pd.DataFrame(neighborhood_scores, index=categories, columns=categories)