Merge pull request #141 from PolicyEngine/nikhilwoodruff/issue140

nikhilwoodruff · web-flow · commit 8096a5bf62ed · 2025-06-24T14:14:33.000+01:00
Fix calibration log
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    fixed:
+    - Missing columns in the calibration log.
diff --git a/policyengine_uk_data/datasets/frs/local_areas/constituencies/calibrate.py b/policyengine_uk_data/datasets/frs/local_areas/constituencies/calibrate.py
@@ -160,6 +160,13 @@ def dropout_weights(weights, p):
                 excluded_training_targets,
             )
             performance_step["epoch"] = epoch
+            performance_step["loss"] = performance_step.rel_abs_error**2
+            performance_step["target_name"] = [
+                f"{area}/{metric}"
+                for area, metric in zip(
+                    performance_step.name, performance_step.metric
+                )
+            ]
             performance = pd.concat(
                 [performance, performance_step], ignore_index=True
             )

Original file line number	Diff line number	Diff line change
`@@ -160,6 +160,13 @@ def dropout_weights(weights, p):`
`160`	`160`	`excluded_training_targets,`
`161`	`161`	`)`
`162`	`162`	`performance_step["epoch"] = epoch`
	`163`	`+ performance_step["loss"] = performance_step.rel_abs_error**2`
	`164`	`+ performance_step["target_name"] = [`
	`165`	`+ f"{area}/{metric}"`
	`166`	`+ for area, metric in zip(`
	`167`	`+ performance_step.name, performance_step.metric`
	`168`	`+ )`
	`169`	`+ ]`
`163`	`170`	`performance = pd.concat(`
`164`	`171`	`[performance, performance_step], ignore_index=True`
`165`	`172`	`)`