tune similarity metric weights

gcrsef-gang · Mar 6, 2023 · 562c6c4 · 562c6c4
1 parent bcd05f3
commit 562c6c4
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/rba/community_generation.py b/rba/community_generation.py
@@ -99,7 +99,7 @@ def compute_precinct_similarities(graph, name=None, verbose=False):
         transformed_log_pop_2 = (data2["log_pop_density"] - min_log_pop_density) / log_pop_density_range
         distances["pop_density"] = abs(transformed_log_pop_1 - transformed_log_pop_2)
 
-        distances["county"] = 1 if data1["COUNTYFP10"] == data2["COUNTYFP10"] else 0
+        distances["county"] = 0 if data1["COUNTYFP10"] == data2["COUNTYFP10"] else 1
 
         distance_list = []
         weight_list = []
@@ -118,7 +118,7 @@ def compute_precinct_similarities(graph, name=None, verbose=False):
 
         if verbose:
             edges_iter.set_description(
-                " ".join([f"{distances[metric]}_dist={round(dist, 3)}" for metric, dist in distances.items()])
+                " ".join([f"{metric}_dist={round(dist, 3)}" for metric, dist in distances.items()])
                 + f" similarity={round(similarity, 3)}"
             )
 

diff --git a/rba/constants.py b/rba/constants.py
@@ -12,10 +12,10 @@
 MINORITY_NAMES = ["black", "hispanic", "asian", "native", "islander"]
 
 SIMILARITY_WEIGHTS = {
-    "race": 1.0,
+    "race": 1.5,
     "votes": 1.0,
     "pop_density": 1.0,
-    "county": 0.5
+    "county": 0.1
 }
 
 # Due logarithm domain redistriction

diff --git a/rba/data/2010/maryland_communities.json b/rba/data/2010/maryland_communities.json