fix: neighborhood self weight (#42)

OzetteTech · Aug 31, 2023 · 8c85c0e · 8c85c0e
1 parent 418a7c9
commit 8c85c0e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)