Merge pull request #4 from radekosmulski/patch-1

tunguz · Jun 18, 2022 · c93403c · c93403c
2 parents 653b722 + 9eb1b2f
commit c93403c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/census_income/XGBClassifier.ipynb b/scripts/census_income/XGBClassifier.ipynb
@@ -238,6 +238,7 @@
    "source": [
     "train_data['gender'] = 1*(train_data['gender'] == ' Male')\n",
     "test_data['gender'] = 1*(test_data['gender'] == ' Male')\n",
+    "test_data['income_bracket'] = test_data['income_bracket'].str[:-1]\n",
     "\n",
     "train_data['native_country'] = 1*(train_data['native_country'] == ' United-States')\n",
     "test_data['native_country'] = 1*(test_data['native_country'] == ' United-States')"