DOC: Do not override DataFrame with array

Fixes #225
sebp · Oct 26, 2021 · ec0ed18 · ec0ed18
1 parent 39a7409
commit ec0ed18
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/doc/user_guide/evaluating-survival-models.ipynb b/doc/user_guide/evaluating-survival-models.ipynb
@@ -354,8 +354,7 @@
     "num_columns = ['age', 'creatinine', 'kappa', 'lambda']\n",
     "\n",
     "imputer = SimpleImputer().fit(x_train.loc[:, num_columns])\n",
-    "x_train = imputer.transform(x_train.loc[:, num_columns])\n",
-    "x_test = imputer.transform(x_test.loc[:, num_columns])"
+    "x_test_imputed = imputer.transform(x_test.loc[:, num_columns])"
    ]
   },
   {
@@ -447,8 +446,8 @@
     "\n",
     "    \n",
     "for i, col in enumerate(num_columns):\n",
-    "    plot_cumulative_dynamic_auc(x_test[:, i], col, color=\"C{}\".format(i))\n",
-    "    ret = concordance_index_ipcw(y_train, y_test, x_test[:, i], tau=times[-1])"
+    "    plot_cumulative_dynamic_auc(x_test_imputed[:, i], col, color=\"C{}\".format(i))\n",
+    "    ret = concordance_index_ipcw(y_train, y_test, x_test_imputed[:, i], tau=times[-1])"
    ]
   },
   {