fix: map class predictions for XGB results

dreamquark-ai · Feb 7, 2020 · 3747e2f · 3747e2f
1 parent d8631f8
commit 3747e2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/forest_example.ipynb b/forest_example.ipynb
@@ -381,11 +381,11 @@
    "outputs": [],
    "source": [
     "preds_valid = np.array(clf_xgb.predict_proba(X_valid, ))\n",
-    "valid_acc = accuracy_score(y_pred=np.argmax(preds_valid, axis=1), y_true=y_valid)\n",
+    "valid_acc = accuracy_score(y_pred=np.argmax(preds_valid, axis=1) + 1, y_true=y_valid)\n",
     "print(valid_acc)\n",
     "\n",
     "preds_test = np.array(clf_xgb.predict_proba(X_test))\n",
-    "test_acc = accuracy_score(y_pred=np.argmax(preds_test, axis=1), y_true=y_test)\n",
+    "test_acc = accuracy_score(y_pred=np.argmax(preds_test, axis=1) + 1, y_true=y_test)\n",
     "print(test_acc)"
    ]
   },