interactions done

BasisResearch · Jun 25, 2024 · 74d6be0 · 74d6be0
1 parent e68d51e
commit 74d6be0
Show file tree

Hide file tree

Showing 7 changed files with 1,195 additions and 219 deletions.
diff --git a/cities/modeling/evaluation.py b/cities/modeling/evaluation.py
@@ -1,5 +1,5 @@
 import os
-
+import pyro
 import matplotlib.pyplot as plt
 import seaborn as sns
 import torch
@@ -44,6 +44,7 @@ def test_performance(
     plot=True,
 ):
 
+    pyro.clear_param_store()
     # TODO perhaps remove the original categorical levels here
 
     assert all(

diff --git a/cities/utils/data_loader.py b/cities/utils/data_loader.py
@@ -45,7 +45,7 @@ def __getitem__(self, idx):
         }
 
 
-def select_from_data(data, kwarg_names: Dict["str", List["str"]]):
+def select_from_data(data, kwarg_names: Dict[str, List[str]]):
     _data = {}
     _data["outcome"] = data["continuous"][kwarg_names["outcome"]]
     _data["categorical"] = {

diff --git a/data/minneapolis/processed/zoning_dataset.pt b/data/minneapolis/processed/zoning_dataset.pt
diff --git a/docs/experimental_notebooks/zoning/performance_tests.ipynb b/docs/experimental_notebooks/zoning/performance_tests.ipynb
diff --git a/docs/experimental_notebooks/zoning/zoning_categorical_interactions.ipynb b/docs/experimental_notebooks/zoning/zoning_categorical_interactions.ipynb
diff --git a/docs/experimental_notebooks/zoning/zoning_data.ipynb b/docs/experimental_notebooks/zoning/zoning_data.ipynb
diff --git a/docs/experimental_notebooks/zoning/zoning_performance_tests.ipynb b/docs/experimental_notebooks/zoning/zoning_performance_tests.ipynb