erjonb commited on
Commit
c3d292e
·
1 Parent(s): 4090835

Upload P2 - Secom Notebook - Mercury.ipynb

Browse files
P2 - Secom Notebook - Mercury.ipynb CHANGED
@@ -46,7 +46,6 @@
46
  "import numpy as np\n",
47
  "import seaborn as sns\n",
48
  "import matplotlib.pyplot as plt\n",
49
- "from scipy import stats\n",
50
  "from sklearn.model_selection import train_test_split\n",
51
  "import mercury as mr"
52
  ]
@@ -1298,11 +1297,6 @@
1298
  },
1299
  "outputs": [],
1300
  "source": [
1301
- "# disable warnings\n",
1302
- "\n",
1303
- "import warnings\n",
1304
- "warnings.filterwarnings('ignore')\n",
1305
- "\n",
1306
  "# train the model using the function train_model and save the predictions as new variables\n",
1307
  "\n",
1308
  "y_pred_random_forest = train_model('random_forest', X_train_res, y_train_res, X_test_imputed, y_test)\n",
 
46
  "import numpy as np\n",
47
  "import seaborn as sns\n",
48
  "import matplotlib.pyplot as plt\n",
 
49
  "from sklearn.model_selection import train_test_split\n",
50
  "import mercury as mr"
51
  ]
 
1297
  },
1298
  "outputs": [],
1299
  "source": [
 
 
 
 
 
1300
  "# train the model using the function train_model and save the predictions as new variables\n",
1301
  "\n",
1302
  "y_pred_random_forest = train_model('random_forest', X_train_res, y_train_res, X_test_imputed, y_test)\n",