Added the mushroom data set into the jupyter notebook
This commit is contained in:
@@ -10,17 +10,282 @@
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"Notebook is ready\n"
|
||||
"Feature data: (8124, 22)\n",
|
||||
"Target data: (8124,)\n"
|
||||
]
|
||||
},
|
||||
{
|
||||
"data": {
|
||||
"text/html": [
|
||||
"<div>\n",
|
||||
"<style scoped>\n",
|
||||
" .dataframe tbody tr th:only-of-type {\n",
|
||||
" vertical-align: middle;\n",
|
||||
" }\n",
|
||||
"\n",
|
||||
" .dataframe tbody tr th {\n",
|
||||
" vertical-align: top;\n",
|
||||
" }\n",
|
||||
"\n",
|
||||
" .dataframe thead th {\n",
|
||||
" text-align: right;\n",
|
||||
" }\n",
|
||||
"</style>\n",
|
||||
"<table border=\"1\" class=\"dataframe\">\n",
|
||||
" <thead>\n",
|
||||
" <tr style=\"text-align: right;\">\n",
|
||||
" <th></th>\n",
|
||||
" <th>cap_shape</th>\n",
|
||||
" <th>cap_surface</th>\n",
|
||||
" <th>cap_color</th>\n",
|
||||
" <th>bruises</th>\n",
|
||||
" <th>odor</th>\n",
|
||||
" <th>gill_attachment</th>\n",
|
||||
" <th>gill_spacing</th>\n",
|
||||
" <th>gill_size</th>\n",
|
||||
" <th>gill_color</th>\n",
|
||||
" <th>stalk_shape</th>\n",
|
||||
" <th>...</th>\n",
|
||||
" <th>stalk_surface_below_ring</th>\n",
|
||||
" <th>stalk_color_above_ring</th>\n",
|
||||
" <th>stalk_color_below_ring</th>\n",
|
||||
" <th>veil_type</th>\n",
|
||||
" <th>veil_color</th>\n",
|
||||
" <th>ring_number</th>\n",
|
||||
" <th>ring_type</th>\n",
|
||||
" <th>spore-print-color</th>\n",
|
||||
" <th>population</th>\n",
|
||||
" <th>habitat</th>\n",
|
||||
" </tr>\n",
|
||||
" </thead>\n",
|
||||
" <tbody>\n",
|
||||
" <tr>\n",
|
||||
" <th>0</th>\n",
|
||||
" <td>x</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>t</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>c</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>k</td>\n",
|
||||
" <td>e</td>\n",
|
||||
" <td>...</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>o</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>k</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>u</td>\n",
|
||||
" </tr>\n",
|
||||
" <tr>\n",
|
||||
" <th>1</th>\n",
|
||||
" <td>x</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>y</td>\n",
|
||||
" <td>t</td>\n",
|
||||
" <td>a</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>c</td>\n",
|
||||
" <td>b</td>\n",
|
||||
" <td>k</td>\n",
|
||||
" <td>e</td>\n",
|
||||
" <td>...</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>o</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>g</td>\n",
|
||||
" </tr>\n",
|
||||
" <tr>\n",
|
||||
" <th>2</th>\n",
|
||||
" <td>b</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>t</td>\n",
|
||||
" <td>l</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>c</td>\n",
|
||||
" <td>b</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>e</td>\n",
|
||||
" <td>...</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>o</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>m</td>\n",
|
||||
" </tr>\n",
|
||||
" <tr>\n",
|
||||
" <th>3</th>\n",
|
||||
" <td>x</td>\n",
|
||||
" <td>y</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>t</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>c</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>e</td>\n",
|
||||
" <td>...</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>o</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>k</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>u</td>\n",
|
||||
" </tr>\n",
|
||||
" <tr>\n",
|
||||
" <th>4</th>\n",
|
||||
" <td>x</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>g</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>f</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>b</td>\n",
|
||||
" <td>k</td>\n",
|
||||
" <td>t</td>\n",
|
||||
" <td>...</td>\n",
|
||||
" <td>s</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>p</td>\n",
|
||||
" <td>w</td>\n",
|
||||
" <td>o</td>\n",
|
||||
" <td>e</td>\n",
|
||||
" <td>n</td>\n",
|
||||
" <td>a</td>\n",
|
||||
" <td>g</td>\n",
|
||||
" </tr>\n",
|
||||
" </tbody>\n",
|
||||
"</table>\n",
|
||||
"<p>5 rows × 22 columns</p>\n",
|
||||
"</div>"
|
||||
],
|
||||
"text/plain": [
|
||||
" cap_shape cap_surface cap_color bruises odor gill_attachment gill_spacing \\\n",
|
||||
"0 x s n t p f c \n",
|
||||
"1 x s y t a f c \n",
|
||||
"2 b s w t l f c \n",
|
||||
"3 x y w t p f c \n",
|
||||
"4 x s g f n f w \n",
|
||||
"\n",
|
||||
" gill_size gill_color stalk_shape ... stalk_surface_below_ring \\\n",
|
||||
"0 n k e ... s \n",
|
||||
"1 b k e ... s \n",
|
||||
"2 b n e ... s \n",
|
||||
"3 n n e ... s \n",
|
||||
"4 b k t ... s \n",
|
||||
"\n",
|
||||
" stalk_color_above_ring stalk_color_below_ring veil_type veil_color \\\n",
|
||||
"0 w w p w \n",
|
||||
"1 w w p w \n",
|
||||
"2 w w p w \n",
|
||||
"3 w w p w \n",
|
||||
"4 w w p w \n",
|
||||
"\n",
|
||||
" ring_number ring_type spore-print-color population habitat \n",
|
||||
"0 o p k s u \n",
|
||||
"1 o p n n g \n",
|
||||
"2 o p n n m \n",
|
||||
"3 o p k s u \n",
|
||||
"4 o e n a g \n",
|
||||
"\n",
|
||||
"[5 rows x 22 columns]"
|
||||
]
|
||||
},
|
||||
"metadata": {},
|
||||
"output_type": "display_data"
|
||||
},
|
||||
{
|
||||
"data": {
|
||||
"text/plain": [
|
||||
"0 p\n",
|
||||
"1 e\n",
|
||||
"2 e\n",
|
||||
"3 p\n",
|
||||
"4 e\n",
|
||||
"Name: class, dtype: category\n",
|
||||
"Categories (2, str): ['e', 'p']"
|
||||
]
|
||||
},
|
||||
"metadata": {},
|
||||
"output_type": "display_data"
|
||||
},
|
||||
{
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"class\n",
|
||||
"e 4208\n",
|
||||
"p 3916\n",
|
||||
"Name: count, dtype: int64\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"import pandas as pd\n",
|
||||
"import sklearn\n",
|
||||
"import shap\n",
|
||||
"import xgboost\n",
|
||||
"from sklearn.datasets import fetch_openml\n",
|
||||
"\n",
|
||||
"print(\"Notebook is ready\")"
|
||||
"mushroom = fetch_openml(data_id=43922, as_frame=True)\n",
|
||||
"\n",
|
||||
"X = mushroom.data\n",
|
||||
"y = mushroom.target\n",
|
||||
"\n",
|
||||
"print(\"Feature data:\", X.shape)\n",
|
||||
"print(\"Target data:\", y.shape)\n",
|
||||
"display(X.head())\n",
|
||||
"display(y.head())\n",
|
||||
"print(y.value_counts())"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 2,
|
||||
"id": "69e1a254",
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"Missing cells: 0\n",
|
||||
"Rows with missing cells: 0\n",
|
||||
"stalk_root\n",
|
||||
"b 3776\n",
|
||||
"? 2480\n",
|
||||
"e 1120\n",
|
||||
"c 556\n",
|
||||
"r 192\n",
|
||||
"Name: count, dtype: int64\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"print(\"Missing cells:\", X.isna().sum().sum())\n",
|
||||
"print(\"Rows with missing cells:\", X.isna().any(axis=1).sum())\n",
|
||||
"print(X[\"stalk_root\"].value_counts(dropna=False))"
|
||||
]
|
||||
}
|
||||
],
|
||||
|
||||
Reference in New Issue
Block a user