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",
|