Remove <10k AGI taxable targets and update documentation

PolicyEngine · Sep 25, 2024 · b5121a8 · b5121a8
1 parent eab8938
commit b5121a8
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 73 deletions.
diff --git a/docs/validation.ipynb b/docs/validation.ipynb
diff --git a/policyengine_us_data/datasets/cps/enhanced_cps.py b/policyengine_us_data/datasets/cps/enhanced_cps.py
@@ -55,7 +55,7 @@ def loss(weights):
 
     start_loss = None
 
-    iterator = trange(10_000)
+    iterator = trange(1_000)
     for i in iterator:
         optimizer.zero_grad()
         l = loss(torch.exp(weights))

diff --git a/policyengine_us_data/utils/loss.py b/policyengine_us_data/utils/loss.py
@@ -71,6 +71,9 @@ def build_loss_matrix(dataset: type, time_period):
         if not row["Taxable only"]:
             continue  # exclude non "taxable returns" statistics
 
+        if row["AGI upper bound"] <= 10_000:
+            continue
+
         mask = (
             (agi >= row["AGI lower bound"])
             * (agi < row["AGI upper bound"])
@@ -180,6 +183,17 @@ def build_loss_matrix(dataset: type, time_period):
             ).calibration.gov.cbo._children[variable_name]
         )
 
+    # Treasury EITC
+
+    loss_matrix["treasury/eitc"] = sim.calculate(
+        "eitc", map_to="household"
+    ).values
+    targets_array.append(
+        sim.tax_benefit_system.parameters(
+            time_period
+        ).calibration.gov.treasury.tax_expenditures.eitc
+    )
+
     # CPS-derived statistics
     # Medical expenses, sum of spm thresholds
     # Child support expenses