Skip to content

Commit 84c71df

Browse files
committed
Fixed error in computing mean correlation with cryus target
1 parent 16f452e commit 84c71df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

target_ensemble.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2977,7 +2977,7 @@
29772977
" )\n",
29782978
" # per era correlation between this target and cyrus\n",
29792979
" mean_corr_with_cryus = validation.groupby(\"era\").apply(\n",
2980-
" lambda d: d[target].corr(d[\"target_cyrus_v4_20\"])\n",
2980+
" lambda d: d[pred_col].corr(d[\"target_cyrus_v4_20\"])\n",
29812981
" ).mean()\n",
29822982
" target_summary_metrics[pred_col].update({\n",
29832983
" \"mean_corr_with_cryus\": mean_corr_with_cryus\n",

0 commit comments

Comments
 (0)