Python/ml/NASA_Asteroids_Classification.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# NASA ASTEROIDS CLASSIFICATION"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Link to the Dataset: [Asteroid Details](https://www.kaggle.com/shrutimehta/nasa-asteroids-classification)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Importing Libraries"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "from sklearn import preprocessing  \n",
    "from statsmodels.stats.outliers_influence import variance_inflation_factor\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.naive_bayes import GaussianNB\n",
    "import matplotlib.pyplot as plt\n",
    "from scipy import stats\n",
    "import seaborn as sns\n",
    "from sklearn.metrics import confusion_matrix"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Getting our Data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "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>Neo Reference ID</th>\n",
       "      <th>Name</th>\n",
       "      <th>Absolute Magnitude</th>\n",
       "      <th>Est Dia in KM(min)</th>\n",
       "      <th>Est Dia in KM(max)</th>\n",
       "      <th>Est Dia in M(min)</th>\n",
       "      <th>Est Dia in M(max)</th>\n",
       "      <th>Est Dia in Miles(min)</th>\n",
       "      <th>Est Dia in Miles(max)</th>\n",
       "      <th>Est Dia in Feet(min)</th>\n",
       "      <th>...</th>\n",
       "      <th>Asc Node Longitude</th>\n",
       "      <th>Orbital Period</th>\n",
       "      <th>Perihelion Distance</th>\n",
       "      <th>Perihelion Arg</th>\n",
       "      <th>Aphelion Dist</th>\n",
       "      <th>Perihelion Time</th>\n",
       "      <th>Mean Anomaly</th>\n",
       "      <th>Mean Motion</th>\n",
       "      <th>Equinox</th>\n",
       "      <th>Hazardous</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>3703080</td>\n",
       "      <td>3703080</td>\n",
       "      <td>21.600</td>\n",
       "      <td>0.127220</td>\n",
       "      <td>0.284472</td>\n",
       "      <td>127.219879</td>\n",
       "      <td>284.472297</td>\n",
       "      <td>0.079051</td>\n",
       "      <td>0.176763</td>\n",
       "      <td>417.388066</td>\n",
       "      <td>...</td>\n",
       "      <td>314.373913</td>\n",
       "      <td>609.599786</td>\n",
       "      <td>0.808259</td>\n",
       "      <td>57.257470</td>\n",
       "      <td>2.005764</td>\n",
       "      <td>2.458162e+06</td>\n",
       "      <td>264.837533</td>\n",
       "      <td>0.590551</td>\n",
       "      <td>J2000</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3723955</td>\n",
       "      <td>3723955</td>\n",
       "      <td>21.300</td>\n",
       "      <td>0.146068</td>\n",
       "      <td>0.326618</td>\n",
       "      <td>146.067964</td>\n",
       "      <td>326.617897</td>\n",
       "      <td>0.090762</td>\n",
       "      <td>0.202951</td>\n",
       "      <td>479.225620</td>\n",
       "      <td>...</td>\n",
       "      <td>136.717242</td>\n",
       "      <td>425.869294</td>\n",
       "      <td>0.718200</td>\n",
       "      <td>313.091975</td>\n",
       "      <td>1.497352</td>\n",
       "      <td>2.457795e+06</td>\n",
       "      <td>173.741112</td>\n",
       "      <td>0.845330</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2446862</td>\n",
       "      <td>2446862</td>\n",
       "      <td>20.300</td>\n",
       "      <td>0.231502</td>\n",
       "      <td>0.517654</td>\n",
       "      <td>231.502122</td>\n",
       "      <td>517.654482</td>\n",
       "      <td>0.143849</td>\n",
       "      <td>0.321655</td>\n",
       "      <td>759.521423</td>\n",
       "      <td>...</td>\n",
       "      <td>259.475979</td>\n",
       "      <td>643.580228</td>\n",
       "      <td>0.950791</td>\n",
       "      <td>248.415038</td>\n",
       "      <td>1.966857</td>\n",
       "      <td>2.458120e+06</td>\n",
       "      <td>292.893654</td>\n",
       "      <td>0.559371</td>\n",
       "      <td>J2000</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3092506</td>\n",
       "      <td>3092506</td>\n",
       "      <td>27.400</td>\n",
       "      <td>0.008801</td>\n",
       "      <td>0.019681</td>\n",
       "      <td>8.801465</td>\n",
       "      <td>19.680675</td>\n",
       "      <td>0.005469</td>\n",
       "      <td>0.012229</td>\n",
       "      <td>28.876199</td>\n",
       "      <td>...</td>\n",
       "      <td>57.173266</td>\n",
       "      <td>514.082140</td>\n",
       "      <td>0.983902</td>\n",
       "      <td>18.707701</td>\n",
       "      <td>1.527904</td>\n",
       "      <td>2.457902e+06</td>\n",
       "      <td>68.741007</td>\n",
       "      <td>0.700277</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3514799</td>\n",
       "      <td>3514799</td>\n",
       "      <td>21.600</td>\n",
       "      <td>0.127220</td>\n",
       "      <td>0.284472</td>\n",
       "      <td>127.219879</td>\n",
       "      <td>284.472297</td>\n",
       "      <td>0.079051</td>\n",
       "      <td>0.176763</td>\n",
       "      <td>417.388066</td>\n",
       "      <td>...</td>\n",
       "      <td>84.629307</td>\n",
       "      <td>495.597821</td>\n",
       "      <td>0.967687</td>\n",
       "      <td>158.263596</td>\n",
       "      <td>1.483543</td>\n",
       "      <td>2.457814e+06</td>\n",
       "      <td>135.142133</td>\n",
       "      <td>0.726395</td>\n",
       "      <td>J2000</td>\n",
       "      <td>True</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4682</th>\n",
       "      <td>3759007</td>\n",
       "      <td>3759007</td>\n",
       "      <td>23.900</td>\n",
       "      <td>0.044112</td>\n",
       "      <td>0.098637</td>\n",
       "      <td>44.111820</td>\n",
       "      <td>98.637028</td>\n",
       "      <td>0.027410</td>\n",
       "      <td>0.061290</td>\n",
       "      <td>144.723824</td>\n",
       "      <td>...</td>\n",
       "      <td>164.183305</td>\n",
       "      <td>457.179984</td>\n",
       "      <td>0.741558</td>\n",
       "      <td>276.395697</td>\n",
       "      <td>1.581299</td>\n",
       "      <td>2.457708e+06</td>\n",
       "      <td>304.306025</td>\n",
       "      <td>0.787436</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4683</th>\n",
       "      <td>3759295</td>\n",
       "      <td>3759295</td>\n",
       "      <td>28.200</td>\n",
       "      <td>0.006089</td>\n",
       "      <td>0.013616</td>\n",
       "      <td>6.089126</td>\n",
       "      <td>13.615700</td>\n",
       "      <td>0.003784</td>\n",
       "      <td>0.008460</td>\n",
       "      <td>19.977449</td>\n",
       "      <td>...</td>\n",
       "      <td>345.225230</td>\n",
       "      <td>407.185767</td>\n",
       "      <td>0.996434</td>\n",
       "      <td>42.111064</td>\n",
       "      <td>1.153835</td>\n",
       "      <td>2.458088e+06</td>\n",
       "      <td>282.978786</td>\n",
       "      <td>0.884117</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4684</th>\n",
       "      <td>3759714</td>\n",
       "      <td>3759714</td>\n",
       "      <td>22.700</td>\n",
       "      <td>0.076658</td>\n",
       "      <td>0.171412</td>\n",
       "      <td>76.657557</td>\n",
       "      <td>171.411509</td>\n",
       "      <td>0.047633</td>\n",
       "      <td>0.106510</td>\n",
       "      <td>251.501180</td>\n",
       "      <td>...</td>\n",
       "      <td>37.026468</td>\n",
       "      <td>690.054279</td>\n",
       "      <td>0.965760</td>\n",
       "      <td>274.692712</td>\n",
       "      <td>2.090708</td>\n",
       "      <td>2.458300e+06</td>\n",
       "      <td>203.501147</td>\n",
       "      <td>0.521698</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4685</th>\n",
       "      <td>3759720</td>\n",
       "      <td>3759720</td>\n",
       "      <td>21.800</td>\n",
       "      <td>0.116026</td>\n",
       "      <td>0.259442</td>\n",
       "      <td>116.025908</td>\n",
       "      <td>259.441818</td>\n",
       "      <td>0.072095</td>\n",
       "      <td>0.161210</td>\n",
       "      <td>380.662441</td>\n",
       "      <td>...</td>\n",
       "      <td>163.802910</td>\n",
       "      <td>662.048343</td>\n",
       "      <td>1.185467</td>\n",
       "      <td>180.346090</td>\n",
       "      <td>1.787733</td>\n",
       "      <td>2.458288e+06</td>\n",
       "      <td>203.524965</td>\n",
       "      <td>0.543767</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4686</th>\n",
       "      <td>3772978</td>\n",
       "      <td>3772978</td>\n",
       "      <td>19.109</td>\n",
       "      <td>0.400641</td>\n",
       "      <td>0.895860</td>\n",
       "      <td>400.640618</td>\n",
       "      <td>895.859655</td>\n",
       "      <td>0.248946</td>\n",
       "      <td>0.556661</td>\n",
       "      <td>1314.437764</td>\n",
       "      <td>...</td>\n",
       "      <td>187.642183</td>\n",
       "      <td>653.679098</td>\n",
       "      <td>0.876110</td>\n",
       "      <td>222.436688</td>\n",
       "      <td>2.071980</td>\n",
       "      <td>2.458319e+06</td>\n",
       "      <td>184.820424</td>\n",
       "      <td>0.550729</td>\n",
       "      <td>J2000</td>\n",
       "      <td>False</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>4687 rows × 40 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      Neo Reference ID     Name  Absolute Magnitude  Est Dia in KM(min)  \\\n",
       "0              3703080  3703080              21.600            0.127220   \n",
       "1              3723955  3723955              21.300            0.146068   \n",
       "2              2446862  2446862              20.300            0.231502   \n",
       "3              3092506  3092506              27.400            0.008801   \n",
       "4              3514799  3514799              21.600            0.127220   \n",
       "...                ...      ...                 ...                 ...   \n",
       "4682           3759007  3759007              23.900            0.044112   \n",
       "4683           3759295  3759295              28.200            0.006089   \n",
       "4684           3759714  3759714              22.700            0.076658   \n",
       "4685           3759720  3759720              21.800            0.116026   \n",
       "4686           3772978  3772978              19.109            0.400641   \n",
       "\n",
       "      Est Dia in KM(max)  Est Dia in M(min)  Est Dia in M(max)  \\\n",
       "0               0.284472         127.219879         284.472297   \n",
       "1               0.326618         146.067964         326.617897   \n",
       "2               0.517654         231.502122         517.654482   \n",
       "3               0.019681           8.801465          19.680675   \n",
       "4               0.284472         127.219879         284.472297   \n",
       "...                  ...                ...                ...   \n",
       "4682            0.098637          44.111820          98.637028   \n",
       "4683            0.013616           6.089126          13.615700   \n",
       "4684            0.171412          76.657557         171.411509   \n",
       "4685            0.259442         116.025908         259.441818   \n",
       "4686            0.895860         400.640618         895.859655   \n",
       "\n",
       "      Est Dia in Miles(min)  Est Dia in Miles(max)  Est Dia in Feet(min)  ...  \\\n",
       "0                  0.079051               0.176763            417.388066  ...   \n",
       "1                  0.090762               0.202951            479.225620  ...   \n",
       "2                  0.143849               0.321655            759.521423  ...   \n",
       "3                  0.005469               0.012229             28.876199  ...   \n",
       "4                  0.079051               0.176763            417.388066  ...   \n",
       "...                     ...                    ...                   ...  ...   \n",
       "4682               0.027410               0.061290            144.723824  ...   \n",
       "4683               0.003784               0.008460             19.977449  ...   \n",
       "4684               0.047633               0.106510            251.501180  ...   \n",
       "4685               0.072095               0.161210            380.662441  ...   \n",
       "4686               0.248946               0.556661           1314.437764  ...   \n",
       "\n",
       "      Asc Node Longitude Orbital Period  Perihelion Distance  Perihelion Arg  \\\n",
       "0             314.373913     609.599786             0.808259       57.257470   \n",
       "1             136.717242     425.869294             0.718200      313.091975   \n",
       "2             259.475979     643.580228             0.950791      248.415038   \n",
       "3              57.173266     514.082140             0.983902       18.707701   \n",
       "4              84.629307     495.597821             0.967687      158.263596   \n",
       "...                  ...            ...                  ...             ...   \n",
       "4682          164.183305     457.179984             0.741558      276.395697   \n",
       "4683          345.225230     407.185767             0.996434       42.111064   \n",
       "4684           37.026468     690.054279             0.965760      274.692712   \n",
       "4685          163.802910     662.048343             1.185467      180.346090   \n",
       "4686          187.642183     653.679098             0.876110      222.436688   \n",
       "\n",
       "      Aphelion Dist  Perihelion Time  Mean Anomaly  Mean Motion  Equinox  \\\n",
       "0          2.005764     2.458162e+06    264.837533     0.590551    J2000   \n",
       "1          1.497352     2.457795e+06    173.741112     0.845330    J2000   \n",
       "2          1.966857     2.458120e+06    292.893654     0.559371    J2000   \n",
       "3          1.527904     2.457902e+06     68.741007     0.700277    J2000   \n",
       "4          1.483543     2.457814e+06    135.142133     0.726395    J2000   \n",
       "...             ...              ...           ...          ...      ...   \n",
       "4682       1.581299     2.457708e+06    304.306025     0.787436    J2000   \n",
       "4683       1.153835     2.458088e+06    282.978786     0.884117    J2000   \n",
       "4684       2.090708     2.458300e+06    203.501147     0.521698    J2000   \n",
       "4685       1.787733     2.458288e+06    203.524965     0.543767    J2000   \n",
       "4686       2.071980     2.458319e+06    184.820424     0.550729    J2000   \n",
       "\n",
       "      Hazardous  \n",
       "0          True  \n",
       "1         False  \n",
       "2          True  \n",
       "3         False  \n",
       "4          True  \n",
       "...         ...  \n",
       "4682      False  \n",
       "4683      False  \n",
       "4684      False  \n",
       "4685      False  \n",
       "4686      False  \n",
       "\n",
       "[4687 rows x 40 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = pd.read_csv(r'nasa.csv', encoding = 'unicode_escape')\n",
    "df"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Data Preprocessing"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 4687 entries, 0 to 4686\n",
      "Data columns (total 40 columns):\n",
      " #   Column                        Non-Null Count  Dtype  \n",
      "---  ------                        --------------  -----  \n",
      " 0   Neo Reference ID              4687 non-null   int64  \n",
      " 1   Name                          4687 non-null   int64  \n",
      " 2   Absolute Magnitude            4687 non-null   float64\n",
      " 3   Est Dia in KM(min)            4687 non-null   float64\n",
      " 4   Est Dia in KM(max)            4687 non-null   float64\n",
      " 5   Est Dia in M(min)             4687 non-null   float64\n",
      " 6   Est Dia in M(max)             4687 non-null   float64\n",
      " 7   Est Dia in Miles(min)         4687 non-null   float64\n",
      " 8   Est Dia in Miles(max)         4687 non-null   float64\n",
      " 9   Est Dia in Feet(min)          4687 non-null   float64\n",
      " 10  Est Dia in Feet(max)          4687 non-null   float64\n",
      " 11  Close Approach Date           4687 non-null   object \n",
      " 12  Epoch Date Close Approach     4687 non-null   int64  \n",
      " 13  Relative Velocity km per sec  4687 non-null   float64\n",
      " 14  Relative Velocity km per hr   4687 non-null   float64\n",
      " 15  Miles per hour                4687 non-null   float64\n",
      " 16  Miss Dist.(Astronomical)      4687 non-null   float64\n",
      " 17  Miss Dist.(lunar)             4687 non-null   float64\n",
      " 18  Miss Dist.(kilometers)        4687 non-null   float64\n",
      " 19  Miss Dist.(miles)             4687 non-null   float64\n",
      " 20  Orbiting Body                 4687 non-null   object \n",
      " 21  Orbit ID                      4687 non-null   int64  \n",
      " 22  Orbit Determination Date      4687 non-null   object \n",
      " 23  Orbit Uncertainity            4687 non-null   int64  \n",
      " 24  Minimum Orbit Intersection    4687 non-null   float64\n",
      " 25  Jupiter Tisserand Invariant   4687 non-null   float64\n",
      " 26  Epoch Osculation              4687 non-null   float64\n",
      " 27  Eccentricity                  4687 non-null   float64\n",
      " 28  Semi Major Axis               4687 non-null   float64\n",
      " 29  Inclination                   4687 non-null   float64\n",
      " 30  Asc Node Longitude            4687 non-null   float64\n",
      " 31  Orbital Period                4687 non-null   float64\n",
      " 32  Perihelion Distance           4687 non-null   float64\n",
      " 33  Perihelion Arg                4687 non-null   float64\n",
      " 34  Aphelion Dist                 4687 non-null   float64\n",
      " 35  Perihelion Time               4687 non-null   float64\n",
      " 36  Mean Anomaly                  4687 non-null   float64\n",
      " 37  Mean Motion                   4687 non-null   float64\n",
      " 38  Equinox                       4687 non-null   object \n",
      " 39  Hazardous                     4687 non-null   bool   \n",
      "dtypes: bool(1), float64(30), int64(5), object(4)\n",
      "memory usage: 1.4+ MB\n"
     ]
    }
   ],
   "source": [
    "df.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "df = df.drop(['Neo Reference ID','Name'], axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "# no null values are present\n",
    "# but, we need to encode some attributes\n",
    "label_encoder = preprocessing.LabelEncoder()  # encoding data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "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>Absolute Magnitude</th>\n",
       "      <th>Est Dia in KM(min)</th>\n",
       "      <th>Est Dia in KM(max)</th>\n",
       "      <th>Est Dia in M(min)</th>\n",
       "      <th>Est Dia in M(max)</th>\n",
       "      <th>Est Dia in Miles(min)</th>\n",
       "      <th>Est Dia in Miles(max)</th>\n",
       "      <th>Est Dia in Feet(min)</th>\n",
       "      <th>Est Dia in Feet(max)</th>\n",
       "      <th>Close Approach Date</th>\n",
       "      <th>...</th>\n",
       "      <th>Asc Node Longitude</th>\n",
       "      <th>Orbital Period</th>\n",
       "      <th>Perihelion Distance</th>\n",
       "      <th>Perihelion Arg</th>\n",
       "      <th>Aphelion Dist</th>\n",
       "      <th>Perihelion Time</th>\n",
       "      <th>Mean Anomaly</th>\n",
       "      <th>Mean Motion</th>\n",
       "      <th>Equinox</th>\n",
       "      <th>Hazardous</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>21.600</td>\n",
       "      <td>0.127220</td>\n",
       "      <td>0.284472</td>\n",
       "      <td>127.219879</td>\n",
       "      <td>284.472297</td>\n",
       "      <td>0.079051</td>\n",
       "      <td>0.176763</td>\n",
       "      <td>417.388066</td>\n",
       "      <td>933.308089</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>314.373913</td>\n",
       "      <td>609.599786</td>\n",
       "      <td>0.808259</td>\n",
       "      <td>57.257470</td>\n",
       "      <td>2.005764</td>\n",
       "      <td>2.458162e+06</td>\n",
       "      <td>264.837533</td>\n",
       "      <td>0.590551</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>21.300</td>\n",
       "      <td>0.146068</td>\n",
       "      <td>0.326618</td>\n",
       "      <td>146.067964</td>\n",
       "      <td>326.617897</td>\n",
       "      <td>0.090762</td>\n",
       "      <td>0.202951</td>\n",
       "      <td>479.225620</td>\n",
       "      <td>1071.581063</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>136.717242</td>\n",
       "      <td>425.869294</td>\n",
       "      <td>0.718200</td>\n",
       "      <td>313.091975</td>\n",
       "      <td>1.497352</td>\n",
       "      <td>2.457795e+06</td>\n",
       "      <td>173.741112</td>\n",
       "      <td>0.845330</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20.300</td>\n",
       "      <td>0.231502</td>\n",
       "      <td>0.517654</td>\n",
       "      <td>231.502122</td>\n",
       "      <td>517.654482</td>\n",
       "      <td>0.143849</td>\n",
       "      <td>0.321655</td>\n",
       "      <td>759.521423</td>\n",
       "      <td>1698.341531</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>259.475979</td>\n",
       "      <td>643.580228</td>\n",
       "      <td>0.950791</td>\n",
       "      <td>248.415038</td>\n",
       "      <td>1.966857</td>\n",
       "      <td>2.458120e+06</td>\n",
       "      <td>292.893654</td>\n",
       "      <td>0.559371</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>27.400</td>\n",
       "      <td>0.008801</td>\n",
       "      <td>0.019681</td>\n",
       "      <td>8.801465</td>\n",
       "      <td>19.680675</td>\n",
       "      <td>0.005469</td>\n",
       "      <td>0.012229</td>\n",
       "      <td>28.876199</td>\n",
       "      <td>64.569144</td>\n",
       "      <td>2</td>\n",
       "      <td>...</td>\n",
       "      <td>57.173266</td>\n",
       "      <td>514.082140</td>\n",
       "      <td>0.983902</td>\n",
       "      <td>18.707701</td>\n",
       "      <td>1.527904</td>\n",
       "      <td>2.457902e+06</td>\n",
       "      <td>68.741007</td>\n",
       "      <td>0.700277</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>21.600</td>\n",
       "      <td>0.127220</td>\n",
       "      <td>0.284472</td>\n",
       "      <td>127.219879</td>\n",
       "      <td>284.472297</td>\n",
       "      <td>0.079051</td>\n",
       "      <td>0.176763</td>\n",
       "      <td>417.388066</td>\n",
       "      <td>933.308089</td>\n",
       "      <td>2</td>\n",
       "      <td>...</td>\n",
       "      <td>84.629307</td>\n",
       "      <td>495.597821</td>\n",
       "      <td>0.967687</td>\n",
       "      <td>158.263596</td>\n",
       "      <td>1.483543</td>\n",
       "      <td>2.457814e+06</td>\n",
       "      <td>135.142133</td>\n",
       "      <td>0.726395</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4682</th>\n",
       "      <td>23.900</td>\n",
       "      <td>0.044112</td>\n",
       "      <td>0.098637</td>\n",
       "      <td>44.111820</td>\n",
       "      <td>98.637028</td>\n",
       "      <td>0.027410</td>\n",
       "      <td>0.061290</td>\n",
       "      <td>144.723824</td>\n",
       "      <td>323.612307</td>\n",
       "      <td>776</td>\n",
       "      <td>...</td>\n",
       "      <td>164.183305</td>\n",
       "      <td>457.179984</td>\n",
       "      <td>0.741558</td>\n",
       "      <td>276.395697</td>\n",
       "      <td>1.581299</td>\n",
       "      <td>2.457708e+06</td>\n",
       "      <td>304.306025</td>\n",
       "      <td>0.787436</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4683</th>\n",
       "      <td>28.200</td>\n",
       "      <td>0.006089</td>\n",
       "      <td>0.013616</td>\n",
       "      <td>6.089126</td>\n",
       "      <td>13.615700</td>\n",
       "      <td>0.003784</td>\n",
       "      <td>0.008460</td>\n",
       "      <td>19.977449</td>\n",
       "      <td>44.670934</td>\n",
       "      <td>776</td>\n",
       "      <td>...</td>\n",
       "      <td>345.225230</td>\n",
       "      <td>407.185767</td>\n",
       "      <td>0.996434</td>\n",
       "      <td>42.111064</td>\n",
       "      <td>1.153835</td>\n",
       "      <td>2.458088e+06</td>\n",
       "      <td>282.978786</td>\n",
       "      <td>0.884117</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4684</th>\n",
       "      <td>22.700</td>\n",
       "      <td>0.076658</td>\n",
       "      <td>0.171412</td>\n",
       "      <td>76.657557</td>\n",
       "      <td>171.411509</td>\n",
       "      <td>0.047633</td>\n",
       "      <td>0.106510</td>\n",
       "      <td>251.501180</td>\n",
       "      <td>562.373736</td>\n",
       "      <td>776</td>\n",
       "      <td>...</td>\n",
       "      <td>37.026468</td>\n",
       "      <td>690.054279</td>\n",
       "      <td>0.965760</td>\n",
       "      <td>274.692712</td>\n",
       "      <td>2.090708</td>\n",
       "      <td>2.458300e+06</td>\n",
       "      <td>203.501147</td>\n",
       "      <td>0.521698</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4685</th>\n",
       "      <td>21.800</td>\n",
       "      <td>0.116026</td>\n",
       "      <td>0.259442</td>\n",
       "      <td>116.025908</td>\n",
       "      <td>259.441818</td>\n",
       "      <td>0.072095</td>\n",
       "      <td>0.161210</td>\n",
       "      <td>380.662441</td>\n",
       "      <td>851.187094</td>\n",
       "      <td>776</td>\n",
       "      <td>...</td>\n",
       "      <td>163.802910</td>\n",
       "      <td>662.048343</td>\n",
       "      <td>1.185467</td>\n",
       "      <td>180.346090</td>\n",
       "      <td>1.787733</td>\n",
       "      <td>2.458288e+06</td>\n",
       "      <td>203.524965</td>\n",
       "      <td>0.543767</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4686</th>\n",
       "      <td>19.109</td>\n",
       "      <td>0.400641</td>\n",
       "      <td>0.895860</td>\n",
       "      <td>400.640618</td>\n",
       "      <td>895.859655</td>\n",
       "      <td>0.248946</td>\n",
       "      <td>0.556661</td>\n",
       "      <td>1314.437764</td>\n",
       "      <td>2939.172192</td>\n",
       "      <td>776</td>\n",
       "      <td>...</td>\n",
       "      <td>187.642183</td>\n",
       "      <td>653.679098</td>\n",
       "      <td>0.876110</td>\n",
       "      <td>222.436688</td>\n",
       "      <td>2.071980</td>\n",
       "      <td>2.458319e+06</td>\n",
       "      <td>184.820424</td>\n",
       "      <td>0.550729</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>4687 rows × 38 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      Absolute Magnitude  Est Dia in KM(min)  Est Dia in KM(max)  \\\n",
       "0                 21.600            0.127220            0.284472   \n",
       "1                 21.300            0.146068            0.326618   \n",
       "2                 20.300            0.231502            0.517654   \n",
       "3                 27.400            0.008801            0.019681   \n",
       "4                 21.600            0.127220            0.284472   \n",
       "...                  ...                 ...                 ...   \n",
       "4682              23.900            0.044112            0.098637   \n",
       "4683              28.200            0.006089            0.013616   \n",
       "4684              22.700            0.076658            0.171412   \n",
       "4685              21.800            0.116026            0.259442   \n",
       "4686              19.109            0.400641            0.895860   \n",
       "\n",
       "      Est Dia in M(min)  Est Dia in M(max)  Est Dia in Miles(min)  \\\n",
       "0            127.219879         284.472297               0.079051   \n",
       "1            146.067964         326.617897               0.090762   \n",
       "2            231.502122         517.654482               0.143849   \n",
       "3              8.801465          19.680675               0.005469   \n",
       "4            127.219879         284.472297               0.079051   \n",
       "...                 ...                ...                    ...   \n",
       "4682          44.111820          98.637028               0.027410   \n",
       "4683           6.089126          13.615700               0.003784   \n",
       "4684          76.657557         171.411509               0.047633   \n",
       "4685         116.025908         259.441818               0.072095   \n",
       "4686         400.640618         895.859655               0.248946   \n",
       "\n",
       "      Est Dia in Miles(max)  Est Dia in Feet(min)  Est Dia in Feet(max)  \\\n",
       "0                  0.176763            417.388066            933.308089   \n",
       "1                  0.202951            479.225620           1071.581063   \n",
       "2                  0.321655            759.521423           1698.341531   \n",
       "3                  0.012229             28.876199             64.569144   \n",
       "4                  0.176763            417.388066            933.308089   \n",
       "...                     ...                   ...                   ...   \n",
       "4682               0.061290            144.723824            323.612307   \n",
       "4683               0.008460             19.977449             44.670934   \n",
       "4684               0.106510            251.501180            562.373736   \n",
       "4685               0.161210            380.662441            851.187094   \n",
       "4686               0.556661           1314.437764           2939.172192   \n",
       "\n",
       "      Close Approach Date  ...  Asc Node Longitude  Orbital Period  \\\n",
       "0                       0  ...          314.373913      609.599786   \n",
       "1                       0  ...          136.717242      425.869294   \n",
       "2                       1  ...          259.475979      643.580228   \n",
       "3                       2  ...           57.173266      514.082140   \n",
       "4                       2  ...           84.629307      495.597821   \n",
       "...                   ...  ...                 ...             ...   \n",
       "4682                  776  ...          164.183305      457.179984   \n",
       "4683                  776  ...          345.225230      407.185767   \n",
       "4684                  776  ...           37.026468      690.054279   \n",
       "4685                  776  ...          163.802910      662.048343   \n",
       "4686                  776  ...          187.642183      653.679098   \n",
       "\n",
       "      Perihelion Distance  Perihelion Arg  Aphelion Dist  Perihelion Time  \\\n",
       "0                0.808259       57.257470       2.005764     2.458162e+06   \n",
       "1                0.718200      313.091975       1.497352     2.457795e+06   \n",
       "2                0.950791      248.415038       1.966857     2.458120e+06   \n",
       "3                0.983902       18.707701       1.527904     2.457902e+06   \n",
       "4                0.967687      158.263596       1.483543     2.457814e+06   \n",
       "...                   ...             ...            ...              ...   \n",
       "4682             0.741558      276.395697       1.581299     2.457708e+06   \n",
       "4683             0.996434       42.111064       1.153835     2.458088e+06   \n",
       "4684             0.965760      274.692712       2.090708     2.458300e+06   \n",
       "4685             1.185467      180.346090       1.787733     2.458288e+06   \n",
       "4686             0.876110      222.436688       2.071980     2.458319e+06   \n",
       "\n",
       "      Mean Anomaly  Mean Motion  Equinox  Hazardous  \n",
       "0       264.837533     0.590551        0          1  \n",
       "1       173.741112     0.845330        0          0  \n",
       "2       292.893654     0.559371        0          1  \n",
       "3        68.741007     0.700277        0          0  \n",
       "4       135.142133     0.726395        0          1  \n",
       "...            ...          ...      ...        ...  \n",
       "4682    304.306025     0.787436        0          0  \n",
       "4683    282.978786     0.884117        0          0  \n",
       "4684    203.501147     0.521698        0          0  \n",
       "4685    203.524965     0.543767        0          0  \n",
       "4686    184.820424     0.550729        0          0  \n",
       "\n",
       "[4687 rows x 38 columns]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "a = ['Close Approach Date','Orbiting Body','Orbit Determination Date','Equinox','Hazardous']\n",
    "for i in a:\n",
    "    df[i] = df[i].astype('|S')\n",
    "    df[i] = label_encoder.fit_transform(df[i])\n",
    "df "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "# now we'll check variance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['Absolute Magnitude', 'Est Dia in KM(min)', 'Est Dia in KM(max)',\n",
       "       'Est Dia in M(min)', 'Est Dia in M(max)', 'Est Dia in Miles(min)',\n",
       "       'Est Dia in Miles(max)', 'Est Dia in Feet(min)', 'Est Dia in Feet(max)',\n",
       "       'Close Approach Date', 'Epoch Date Close Approach',\n",
       "       'Relative Velocity km per sec', 'Relative Velocity km per hr',\n",
       "       'Miles per hour', 'Miss Dist.(Astronomical)', 'Miss Dist.(lunar)',\n",
       "       'Miss Dist.(kilometers)', 'Miss Dist.(miles)', 'Orbiting Body',\n",
       "       'Orbit ID', 'Orbit Determination Date', 'Orbit Uncertainity',\n",
       "       'Minimum Orbit Intersection', 'Jupiter Tisserand Invariant',\n",
       "       'Epoch Osculation', 'Eccentricity', 'Semi Major Axis', 'Inclination',\n",
       "       'Asc Node Longitude', 'Orbital Period', 'Perihelion Distance',\n",
       "       'Perihelion Arg', 'Aphelion Dist', 'Perihelion Time', 'Mean Anomaly',\n",
       "       'Mean Motion', 'Equinox', 'Hazardous'],\n",
       "      dtype='object')"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\DELL\\anaconda3\\lib\\site-packages\\statsmodels\\stats\\outliers_influence.py:193: RuntimeWarning: divide by zero encountered in double_scalars\n",
      "  vif = 1. / (1. - r_squared_i)\n",
      "C:\\Users\\DELL\\anaconda3\\lib\\site-packages\\statsmodels\\regression\\linear_model.py:1715: RuntimeWarning: invalid value encountered in double_scalars\n",
      "  return 1 - self.ssr/self.centered_tss\n"
     ]
    }
   ],
   "source": [
    "# checking variance\n",
    "variables = df[['Absolute Magnitude', 'Est Dia in KM(min)', 'Est Dia in KM(max)',\n",
    "       'Est Dia in M(min)', 'Est Dia in M(max)', 'Est Dia in Miles(min)',\n",
    "       'Est Dia in Miles(max)', 'Est Dia in Feet(min)', 'Est Dia in Feet(max)',\n",
    "       'Close Approach Date', 'Epoch Date Close Approach',\n",
    "       'Relative Velocity km per sec', 'Relative Velocity km per hr',\n",
    "       'Miles per hour', 'Miss Dist.(Astronomical)', 'Miss Dist.(lunar)',\n",
    "       'Miss Dist.(kilometers)', 'Miss Dist.(miles)', 'Orbiting Body',\n",
    "       'Orbit ID', 'Orbit Determination Date', 'Orbit Uncertainity',\n",
    "       'Minimum Orbit Intersection', 'Jupiter Tisserand Invariant',\n",
    "       'Epoch Osculation', 'Eccentricity', 'Semi Major Axis', 'Inclination',\n",
    "       'Asc Node Longitude', 'Orbital Period', 'Perihelion Distance',\n",
    "       'Perihelion Arg', 'Aphelion Dist', 'Perihelion Time', 'Mean Anomaly',\n",
    "       'Mean Motion', 'Equinox']]\n",
    "vif = pd.DataFrame()\n",
    "vif['VIF'] = [variance_inflation_factor(variables.values, i) for i in range(variables.shape[1])]\n",
    "vif['Features'] = variables.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "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>VIF</th>\n",
       "      <th>Features</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>4.340988e+00</td>\n",
       "      <td>Absolute Magnitude</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in KM(min)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in KM(max)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in M(min)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in M(max)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in Miles(min)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in Miles(max)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in Feet(min)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>inf</td>\n",
       "      <td>Est Dia in Feet(max)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>6.167477e+04</td>\n",
       "      <td>Close Approach Date</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>2.248939e+06</td>\n",
       "      <td>Epoch Date Close Approach</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>inf</td>\n",
       "      <td>Relative Velocity km per sec</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>inf</td>\n",
       "      <td>Relative Velocity km per hr</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>inf</td>\n",
       "      <td>Miles per hour</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>2.905548e+14</td>\n",
       "      <td>Miss Dist.(Astronomical)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>2.370316e+14</td>\n",
       "      <td>Miss Dist.(lunar)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>1.801440e+14</td>\n",
       "      <td>Miss Dist.(kilometers)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>1.801440e+14</td>\n",
       "      <td>Miss Dist.(miles)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>NaN</td>\n",
       "      <td>Orbiting Body</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>2.755150e+00</td>\n",
       "      <td>Orbit ID</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>1.455896e+00</td>\n",
       "      <td>Orbit Determination Date</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>2.450006e+00</td>\n",
       "      <td>Orbit Uncertainity</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>2.149192e+00</td>\n",
       "      <td>Minimum Orbit Intersection</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>1.281509e+03</td>\n",
       "      <td>Jupiter Tisserand Invariant</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>2.638449e+01</td>\n",
       "      <td>Epoch Osculation</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>1.901452e+01</td>\n",
       "      <td>Eccentricity</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>inf</td>\n",
       "      <td>Semi Major Axis</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>5.303596e+00</td>\n",
       "      <td>Inclination</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>1.008488e+00</td>\n",
       "      <td>Asc Node Longitude</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>1.037250e+03</td>\n",
       "      <td>Orbital Period</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>1.000800e+15</td>\n",
       "      <td>Perihelion Distance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>1.012231e+00</td>\n",
       "      <td>Perihelion Arg</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>2.251800e+15</td>\n",
       "      <td>Aphelion Dist</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>2.768610e+01</td>\n",
       "      <td>Perihelion Time</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>1.238800e+00</td>\n",
       "      <td>Mean Anomaly</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>7.960731e+02</td>\n",
       "      <td>Mean Motion</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>NaN</td>\n",
       "      <td>Equinox</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "             VIF                      Features\n",
       "0   4.340988e+00            Absolute Magnitude\n",
       "1            inf            Est Dia in KM(min)\n",
       "2            inf            Est Dia in KM(max)\n",
       "3            inf             Est Dia in M(min)\n",
       "4            inf             Est Dia in M(max)\n",
       "5            inf         Est Dia in Miles(min)\n",
       "6            inf         Est Dia in Miles(max)\n",
       "7            inf          Est Dia in Feet(min)\n",
       "8            inf          Est Dia in Feet(max)\n",
       "9   6.167477e+04           Close Approach Date\n",
       "10  2.248939e+06     Epoch Date Close Approach\n",
       "11           inf  Relative Velocity km per sec\n",
       "12           inf   Relative Velocity km per hr\n",
       "13           inf                Miles per hour\n",
       "14  2.905548e+14      Miss Dist.(Astronomical)\n",
       "15  2.370316e+14             Miss Dist.(lunar)\n",
       "16  1.801440e+14        Miss Dist.(kilometers)\n",
       "17  1.801440e+14             Miss Dist.(miles)\n",
       "18           NaN                 Orbiting Body\n",
       "19  2.755150e+00                      Orbit ID\n",
       "20  1.455896e+00      Orbit Determination Date\n",
       "21  2.450006e+00            Orbit Uncertainity\n",
       "22  2.149192e+00    Minimum Orbit Intersection\n",
       "23  1.281509e+03   Jupiter Tisserand Invariant\n",
       "24  2.638449e+01              Epoch Osculation\n",
       "25  1.901452e+01                  Eccentricity\n",
       "26           inf               Semi Major Axis\n",
       "27  5.303596e+00                   Inclination\n",
       "28  1.008488e+00            Asc Node Longitude\n",
       "29  1.037250e+03                Orbital Period\n",
       "30  1.000800e+15           Perihelion Distance\n",
       "31  1.012231e+00                Perihelion Arg\n",
       "32  2.251800e+15                 Aphelion Dist\n",
       "33  2.768610e+01               Perihelion Time\n",
       "34  1.238800e+00                  Mean Anomaly\n",
       "35  7.960731e+02                   Mean Motion\n",
       "36           NaN                       Equinox"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "vif"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "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>Absolute Magnitude</th>\n",
       "      <th>Close Approach Date</th>\n",
       "      <th>Epoch Date Close Approach</th>\n",
       "      <th>Miss Dist.(Astronomical)</th>\n",
       "      <th>Miss Dist.(lunar)</th>\n",
       "      <th>Miss Dist.(kilometers)</th>\n",
       "      <th>Miss Dist.(miles)</th>\n",
       "      <th>Orbiting Body</th>\n",
       "      <th>Orbit ID</th>\n",
       "      <th>Orbit Determination Date</th>\n",
       "      <th>...</th>\n",
       "      <th>Asc Node Longitude</th>\n",
       "      <th>Orbital Period</th>\n",
       "      <th>Perihelion Distance</th>\n",
       "      <th>Perihelion Arg</th>\n",
       "      <th>Aphelion Dist</th>\n",
       "      <th>Perihelion Time</th>\n",
       "      <th>Mean Anomaly</th>\n",
       "      <th>Mean Motion</th>\n",
       "      <th>Equinox</th>\n",
       "      <th>Hazardous</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>21.600</td>\n",
       "      <td>0</td>\n",
       "      <td>788947200000</td>\n",
       "      <td>0.419483</td>\n",
       "      <td>163.178711</td>\n",
       "      <td>6.275369e+07</td>\n",
       "      <td>3.899334e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>17</td>\n",
       "      <td>534</td>\n",
       "      <td>...</td>\n",
       "      <td>314.373913</td>\n",
       "      <td>609.599786</td>\n",
       "      <td>0.808259</td>\n",
       "      <td>57.257470</td>\n",
       "      <td>2.005764</td>\n",
       "      <td>2.458162e+06</td>\n",
       "      <td>264.837533</td>\n",
       "      <td>0.590551</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>21.300</td>\n",
       "      <td>0</td>\n",
       "      <td>788947200000</td>\n",
       "      <td>0.383014</td>\n",
       "      <td>148.992630</td>\n",
       "      <td>5.729815e+07</td>\n",
       "      <td>3.560342e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>21</td>\n",
       "      <td>432</td>\n",
       "      <td>...</td>\n",
       "      <td>136.717242</td>\n",
       "      <td>425.869294</td>\n",
       "      <td>0.718200</td>\n",
       "      <td>313.091975</td>\n",
       "      <td>1.497352</td>\n",
       "      <td>2.457795e+06</td>\n",
       "      <td>173.741112</td>\n",
       "      <td>0.845330</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20.300</td>\n",
       "      <td>1</td>\n",
       "      <td>789552000000</td>\n",
       "      <td>0.050956</td>\n",
       "      <td>19.821890</td>\n",
       "      <td>7.622912e+06</td>\n",
       "      <td>4.736658e+06</td>\n",
       "      <td>0</td>\n",
       "      <td>22</td>\n",
       "      <td>1910</td>\n",
       "      <td>...</td>\n",
       "      <td>259.475979</td>\n",
       "      <td>643.580228</td>\n",
       "      <td>0.950791</td>\n",
       "      <td>248.415038</td>\n",
       "      <td>1.966857</td>\n",
       "      <td>2.458120e+06</td>\n",
       "      <td>292.893654</td>\n",
       "      <td>0.559371</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>27.400</td>\n",
       "      <td>2</td>\n",
       "      <td>790156800000</td>\n",
       "      <td>0.285322</td>\n",
       "      <td>110.990387</td>\n",
       "      <td>4.268362e+07</td>\n",
       "      <td>2.652237e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>7</td>\n",
       "      <td>1761</td>\n",
       "      <td>...</td>\n",
       "      <td>57.173266</td>\n",
       "      <td>514.082140</td>\n",
       "      <td>0.983902</td>\n",
       "      <td>18.707701</td>\n",
       "      <td>1.527904</td>\n",
       "      <td>2.457902e+06</td>\n",
       "      <td>68.741007</td>\n",
       "      <td>0.700277</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>21.600</td>\n",
       "      <td>2</td>\n",
       "      <td>790156800000</td>\n",
       "      <td>0.407832</td>\n",
       "      <td>158.646713</td>\n",
       "      <td>6.101082e+07</td>\n",
       "      <td>3.791037e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>25</td>\n",
       "      <td>1190</td>\n",
       "      <td>...</td>\n",
       "      <td>84.629307</td>\n",
       "      <td>495.597821</td>\n",
       "      <td>0.967687</td>\n",
       "      <td>158.263596</td>\n",
       "      <td>1.483543</td>\n",
       "      <td>2.457814e+06</td>\n",
       "      <td>135.142133</td>\n",
       "      <td>0.726395</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4682</th>\n",
       "      <td>23.900</td>\n",
       "      <td>776</td>\n",
       "      <td>1473318000000</td>\n",
       "      <td>0.041361</td>\n",
       "      <td>16.089413</td>\n",
       "      <td>6.187511e+06</td>\n",
       "      <td>3.844741e+06</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>166</td>\n",
       "      <td>...</td>\n",
       "      <td>164.183305</td>\n",
       "      <td>457.179984</td>\n",
       "      <td>0.741558</td>\n",
       "      <td>276.395697</td>\n",
       "      <td>1.581299</td>\n",
       "      <td>2.457708e+06</td>\n",
       "      <td>304.306025</td>\n",
       "      <td>0.787436</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4683</th>\n",
       "      <td>28.200</td>\n",
       "      <td>776</td>\n",
       "      <td>1473318000000</td>\n",
       "      <td>0.006469</td>\n",
       "      <td>2.516399</td>\n",
       "      <td>9.677324e+05</td>\n",
       "      <td>6.013211e+05</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>163</td>\n",
       "      <td>...</td>\n",
       "      <td>345.225230</td>\n",
       "      <td>407.185767</td>\n",
       "      <td>0.996434</td>\n",
       "      <td>42.111064</td>\n",
       "      <td>1.153835</td>\n",
       "      <td>2.458088e+06</td>\n",
       "      <td>282.978786</td>\n",
       "      <td>0.884117</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4684</th>\n",
       "      <td>22.700</td>\n",
       "      <td>776</td>\n",
       "      <td>1473318000000</td>\n",
       "      <td>0.061009</td>\n",
       "      <td>23.732393</td>\n",
       "      <td>9.126775e+06</td>\n",
       "      <td>5.671115e+06</td>\n",
       "      <td>0</td>\n",
       "      <td>17</td>\n",
       "      <td>159</td>\n",
       "      <td>...</td>\n",
       "      <td>37.026468</td>\n",
       "      <td>690.054279</td>\n",
       "      <td>0.965760</td>\n",
       "      <td>274.692712</td>\n",
       "      <td>2.090708</td>\n",
       "      <td>2.458300e+06</td>\n",
       "      <td>203.501147</td>\n",
       "      <td>0.521698</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4685</th>\n",
       "      <td>21.800</td>\n",
       "      <td>776</td>\n",
       "      <td>1473318000000</td>\n",
       "      <td>0.260760</td>\n",
       "      <td>101.435493</td>\n",
       "      <td>3.900908e+07</td>\n",
       "      <td>2.423912e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>6</td>\n",
       "      <td>158</td>\n",
       "      <td>...</td>\n",
       "      <td>163.802910</td>\n",
       "      <td>662.048343</td>\n",
       "      <td>1.185467</td>\n",
       "      <td>180.346090</td>\n",
       "      <td>1.787733</td>\n",
       "      <td>2.458288e+06</td>\n",
       "      <td>203.524965</td>\n",
       "      <td>0.543767</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4686</th>\n",
       "      <td>19.109</td>\n",
       "      <td>776</td>\n",
       "      <td>1473318000000</td>\n",
       "      <td>0.462372</td>\n",
       "      <td>179.862671</td>\n",
       "      <td>6.916986e+07</td>\n",
       "      <td>4.298016e+07</td>\n",
       "      <td>0</td>\n",
       "      <td>13</td>\n",
       "      <td>2275</td>\n",
       "      <td>...</td>\n",
       "      <td>187.642183</td>\n",
       "      <td>653.679098</td>\n",
       "      <td>0.876110</td>\n",
       "      <td>222.436688</td>\n",
       "      <td>2.071980</td>\n",
       "      <td>2.458319e+06</td>\n",
       "      <td>184.820424</td>\n",
       "      <td>0.550729</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>4687 rows × 26 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      Absolute Magnitude  Close Approach Date  Epoch Date Close Approach  \\\n",
       "0                 21.600                    0               788947200000   \n",
       "1                 21.300                    0               788947200000   \n",
       "2                 20.300                    1               789552000000   \n",
       "3                 27.400                    2               790156800000   \n",
       "4                 21.600                    2               790156800000   \n",
       "...                  ...                  ...                        ...   \n",
       "4682              23.900                  776              1473318000000   \n",
       "4683              28.200                  776              1473318000000   \n",
       "4684              22.700                  776              1473318000000   \n",
       "4685              21.800                  776              1473318000000   \n",
       "4686              19.109                  776              1473318000000   \n",
       "\n",
       "      Miss Dist.(Astronomical)  Miss Dist.(lunar)  Miss Dist.(kilometers)  \\\n",
       "0                     0.419483         163.178711            6.275369e+07   \n",
       "1                     0.383014         148.992630            5.729815e+07   \n",
       "2                     0.050956          19.821890            7.622912e+06   \n",
       "3                     0.285322         110.990387            4.268362e+07   \n",
       "4                     0.407832         158.646713            6.101082e+07   \n",
       "...                        ...                ...                     ...   \n",
       "4682                  0.041361          16.089413            6.187511e+06   \n",
       "4683                  0.006469           2.516399            9.677324e+05   \n",
       "4684                  0.061009          23.732393            9.126775e+06   \n",
       "4685                  0.260760         101.435493            3.900908e+07   \n",
       "4686                  0.462372         179.862671            6.916986e+07   \n",
       "\n",
       "      Miss Dist.(miles)  Orbiting Body  Orbit ID  Orbit Determination Date  \\\n",
       "0          3.899334e+07              0        17                       534   \n",
       "1          3.560342e+07              0        21                       432   \n",
       "2          4.736658e+06              0        22                      1910   \n",
       "3          2.652237e+07              0         7                      1761   \n",
       "4          3.791037e+07              0        25                      1190   \n",
       "...                 ...            ...       ...                       ...   \n",
       "4682       3.844741e+06              0         4                       166   \n",
       "4683       6.013211e+05              0         2                       163   \n",
       "4684       5.671115e+06              0        17                       159   \n",
       "4685       2.423912e+07              0         6                       158   \n",
       "4686       4.298016e+07              0        13                      2275   \n",
       "\n",
       "      ...  Asc Node Longitude  Orbital Period  Perihelion Distance  \\\n",
       "0     ...          314.373913      609.599786             0.808259   \n",
       "1     ...          136.717242      425.869294             0.718200   \n",
       "2     ...          259.475979      643.580228             0.950791   \n",
       "3     ...           57.173266      514.082140             0.983902   \n",
       "4     ...           84.629307      495.597821             0.967687   \n",
       "...   ...                 ...             ...                  ...   \n",
       "4682  ...          164.183305      457.179984             0.741558   \n",
       "4683  ...          345.225230      407.185767             0.996434   \n",
       "4684  ...           37.026468      690.054279             0.965760   \n",
       "4685  ...          163.802910      662.048343             1.185467   \n",
       "4686  ...          187.642183      653.679098             0.876110   \n",
       "\n",
       "      Perihelion Arg  Aphelion Dist  Perihelion Time  Mean Anomaly  \\\n",
       "0          57.257470       2.005764     2.458162e+06    264.837533   \n",
       "1         313.091975       1.497352     2.457795e+06    173.741112   \n",
       "2         248.415038       1.966857     2.458120e+06    292.893654   \n",
       "3          18.707701       1.527904     2.457902e+06     68.741007   \n",
       "4         158.263596       1.483543     2.457814e+06    135.142133   \n",
       "...              ...            ...              ...           ...   \n",
       "4682      276.395697       1.581299     2.457708e+06    304.306025   \n",
       "4683       42.111064       1.153835     2.458088e+06    282.978786   \n",
       "4684      274.692712       2.090708     2.458300e+06    203.501147   \n",
       "4685      180.346090       1.787733     2.458288e+06    203.524965   \n",
       "4686      222.436688       2.071980     2.458319e+06    184.820424   \n",
       "\n",
       "      Mean Motion  Equinox  Hazardous  \n",
       "0        0.590551        0          1  \n",
       "1        0.845330        0          0  \n",
       "2        0.559371        0          1  \n",
       "3        0.700277        0          0  \n",
       "4        0.726395        0          1  \n",
       "...           ...      ...        ...  \n",
       "4682     0.787436        0          0  \n",
       "4683     0.884117        0          0  \n",
       "4684     0.521698        0          0  \n",
       "4685     0.543767        0          0  \n",
       "4686     0.550729        0          0  \n",
       "\n",
       "[4687 rows x 26 columns]"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# dropping all columns having vif>10\n",
    "df = df.drop(['Est Dia in KM(min)', 'Est Dia in KM(max)', 'Est Dia in M(min)', 'Est Dia in M(max)', 'Est Dia in Miles(min)',\n",
    "              'Est Dia in Miles(max)', 'Est Dia in Feet(min)', 'Est Dia in Feet(max)', 'Relative Velocity km per sec', \n",
    "              'Relative Velocity km per hr', 'Miles per hour', 'Semi Major Axis'], axis = 1)\n",
    "df"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Data Visualization"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<AxesSubplot:xlabel='Hazardous', ylabel='count'>"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAYsAAAEGCAYAAACUzrmNAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjMuMiwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8vihELAAAACXBIWXMAAAsTAAALEwEAmpwYAAAVaklEQVR4nO3df6zdd33f8ecLB5JskDZRbjLja+aIGTYnDKNceVGRNkpK42UrDrQgI5G4HZVRZCaY6KpkfwzoZIlp/CiBJpMpaWxG8bwBjcdIt+CSUroE9wZCHDuk8eosudiLL7CWMBWvdt7743xuc7CP7/ck+Jx7nft8SF+d7/f9/Xy+53Ms2y99f6eqkCRpPi9Y6AFIkhY/w0KS1MmwkCR1MiwkSZ0MC0lSp3MWegCjcvHFF9eqVasWehiSdFa5//77v1tVEyfXn7dhsWrVKqanpxd6GJJ0VknyvwbVPQwlSepkWEiSOo08LJIsS/LNJF9syxcluTvJo+3zwr62Nyc5mOSRJNf01a9Msq+tuyVJRj1uSdIzxrFn8W7g4b7lm4A9VbUa2NOWSbIG2AhcDqwHbk2yrPW5DdgMrG7T+jGMW5LUjDQskkwC/wT47b7yBmB7m98OXNdX31lVx6rqEHAQWJdkOXBBVd1bvQdZ7ejrI0kag1HvWfwm8OvA0321S6vqCED7vKTVVwBP9LWbabUVbf7k+imSbE4ynWR6dnb2jPwASdIIwyLJPwWOVtX9w3YZUKt56qcWq7ZV1VRVTU1MnHKZsCTpORrlfRavBd6Y5FrgPOCCJP8BeDLJ8qo60g4xHW3tZ4CVff0ngcOtPjmgLkkak5HtWVTVzVU1WVWr6J24/oOqejuwG9jUmm0C7mzzu4GNSc5Nchm9E9l726Gqp5Jc1a6CuqGvjyRpDBbiDu4PAruSvAN4HHgLQFXtT7ILOAAcB7ZU1YnW50bgDuB84K42jdSV/3LHqL9CZ6H7/90NCz0EaUGMJSyq6h7gnjb/PeDq07TbCmwdUJ8GrhjdCCVJ8/EObklSJ8NCktTJsJAkdTIsJEmdDAtJUifDQpLUybCQJHUyLCRJnQwLSVInw0KS1MmwkCR1MiwkSZ0MC0lSJ8NCktTJsJAkdTIsJEmdDAtJUqeRhUWS85LsTfKtJPuTfKDV35/kO0keaNO1fX1uTnIwySNJrumrX5lkX1t3S3sXtyRpTEb5WtVjwOur6odJXgh8Lcncu7M/WlUf6m+cZA2wEbgceCnw5SSvaO/hvg3YDNwHfAlYzxjewy1J6hnZnkX1/LAtvrBNNU+XDcDOqjpWVYeAg8C6JMuBC6rq3qoqYAdw3ajGLUk61UjPWSRZluQB4Chwd1V9va16V5IHk9ye5MJWWwE80dd9ptVWtPmT64O+b3OS6STTs7OzZ/KnSNKSNtKwqKoTVbUWmKS3l3AFvUNKLwfWAkeAD7fmg85D1Dz1Qd+3raqmqmpqYmLiJxy9JGnOWK6Gqqo/B+4B1lfVky1EngY+CaxrzWaAlX3dJoHDrT45oC5JGpNRXg01keSn2/z5wM8B327nIOa8CXioze8GNiY5N8llwGpgb1UdAZ5KclW7CuoG4M5RjVuSdKpRXg21HNieZBm9UNpVVV9M8ukka+kdSnoMeCdAVe1Psgs4ABwHtrQroQBuBO4Azqd3FZRXQknSGI0sLKrqQeA1A+rXz9NnK7B1QH0auOKMDlCSNDTv4JYkdTIsJEmdDAtJUifDQpLUybCQJHUyLCRJnQwLSVInw0KS1MmwkCR1MiwkSZ0MC0lSJ8NCktTJsJAkdTIsJEmdDAtJUifDQpLUybCQJHUyLCRJnUYWFknOS7I3ybeS7E/ygVa/KMndSR5tnxf29bk5ycEkjyS5pq9+ZZJ9bd0tSTKqcUuSTjXKPYtjwOur6tXAWmB9kquAm4A9VbUa2NOWSbIG2AhcDqwHbk2yrG3rNmAzsLpN60c4bknSSUYWFtXzw7b4wjYVsAHY3urbgeva/AZgZ1Udq6pDwEFgXZLlwAVVdW9VFbCjr48kaQxGes4iybIkDwBHgbur6uvApVV1BKB9XtKarwCe6Os+02or2vzJ9UHftznJdJLp2dnZM/pbJGkpG2lYVNWJqloLTNLbS7hinuaDzkPUPPVB37etqqaqampiYuJZj1eSNNhYroaqqj8H7qF3ruHJdmiJ9nm0NZsBVvZ1mwQOt/rkgLokaUxGeTXURJKfbvPnAz8HfBvYDWxqzTYBd7b53cDGJOcmuYzeiey97VDVU0mualdB3dDXR5I0BueMcNvLge3tiqYXALuq6otJ7gV2JXkH8DjwFoCq2p9kF3AAOA5sqaoTbVs3AncA5wN3tUmSNCYjC4uqehB4zYD694CrT9NnK7B1QH0amO98hyRphLyDW5LUybCQJHUyLCRJnQwLSVInw0KS1MmwkCR1MiwkSZ0MC0lSJ8NCktTJsJAkdTIsJEmdDAtJUifDQpLUybCQJHUyLCRJnQwLSVInw0KS1GmU7+BemeQrSR5Osj/Ju1v9/Um+k+SBNl3b1+fmJAeTPJLkmr76lUn2tXW3tHdxS5LGZJTv4D4OvLeqvpHkJcD9Se5u6z5aVR/qb5xkDbARuBx4KfDlJK9o7+G+DdgM3Ad8CViP7+GWpLEZ2Z5FVR2pqm+0+aeAh4EV83TZAOysqmNVdQg4CKxLshy4oKruraoCdgDXjWrckqRTjeWcRZJVwGuAr7fSu5I8mOT2JBe22grgib5uM622os2fXB/0PZuTTCeZnp2dPZM/QZKWtJGHRZIXA58D3lNVP6B3SOnlwFrgCPDhuaYDutc89VOLVduqaqqqpiYmJn7SoUuSmpGGRZIX0guKz1TV5wGq6smqOlFVTwOfBNa15jPAyr7uk8DhVp8cUJckjckor4YK8Cng4ar6SF99eV+zNwEPtfndwMYk5ya5DFgN7K2qI8BTSa5q27wBuHNU45YknWqUV0O9Frge2JfkgVb7V8DbkqyldyjpMeCdAFW1P8ku4AC9K6m2tCuhAG4E7gDOp3cVlFdCSdIYjSwsquprDD7f8KV5+mwFtg6oTwNXnLnRSZKeDe/gliR1MiwkSZ0MC0lSJ8NCktTJsJAkdTIsJEmdhgqLJHuGqUmSnp/mvc8iyXnA3wAubg/8m7tv4gJ6jxGXJC0BXTflvRN4D71guJ9nwuIHwG+NbliSpMVk3rCoqo8BH0vyz6vq42MakyRpkRnqcR9V9fEkPwOs6u9TVTtGNC5J0iIyVFgk+TS9d1A8AMw93G/urXWSpOe5YR8kOAWsaa81lSQtMcPeZ/EQ8LdGORBJ0uI17J7FxcCBJHuBY3PFqnrjSEYlSVpUhg2L949yEJKkxW3Yq6H+cNQDkSQtXsM+7uOpJD9o04+SnEjyg44+K5N8JcnDSfYneXerX5Tk7iSPts8L+/rcnORgkkeSXNNXvzLJvrbulvYubknSmAwVFlX1kqq6oE3nAb8IfKKj23HgvVX194CrgC1J1gA3AXuqajWwpy3T1m0ELgfWA7cmWda2dRuwGVjdpvXP4jdKkn5Cz+mps1X1e8DrO9ocqapvtPmngIeBFcAGYHtrth24rs1vAHZW1bGqOgQcBNYlWQ5cUFX3tkt3d/T1kSSNwbA35b25b/EF9O67GPqeiySrgNcAXwcuraoj0AuUJJe0ZiuA+/q6zbTaX7X5k+uDvmczvT0QXvaylw07PElSh2GvhvqFvvnjwGP09gQ6JXkx8DngPVX1g3lONwxaUfPUTy1WbQO2AUxNTXkDoSSdIcNeDfUrz2XjSV5ILyg+U1Wfb+UnkyxvexXLgaOtPgOs7Os+CRxu9ckBdUnSmAx7NdRkki8kOZrkySSfSzLZ0SfAp4CHq+ojfat2A5va/Cbgzr76xiTnJrmM3onsve2Q1VNJrmrbvKGvjyRpDIY9wf079P4zfym98wX/pdXm81rgeuD1SR5o07XAB4E3JHkUeENbpqr2A7uAA8DvA1uqau6hhTcCv03vpPf/BO4actySpDNg2HMWE1XVHw53JHnPfB2q6msMPt8AcPVp+mwFtg6oTwNXDDdUSdKZNuyexXeTvD3Jsja9HfjeKAcmSVo8hg2Lfwa8FfjfwBHgl4DndNJbknT2GfYw1L8BNlXV/4HeIzuAD9ELEUnS89ywexZ/fy4oAKrq+/RuspMkLQHDhsULTnrg30UMv1ciSTrLDfsf/oeB/5HkP9O7e/qtDLhqSZL0/DTsHdw7kkzTe3hggDdX1YGRjkyStGgMfSiphYMBIUlL0HN6RLkkaWkxLCRJnQwLSVInw0KS1MmwkCR1MiwkSZ0MC0lSJ8NCktTJsJAkdTIsJEmdRhYWSW5PcjTJQ3219yf5zknv5J5bd3OSg0keSXJNX/3KJPvauluSnO5VrZKkERnlnsUdwPoB9Y9W1do2fQkgyRpgI3B563NrkmWt/W3AZmB1mwZtU5I0QiMLi6r6KvD9IZtvAHZW1bGqOgQcBNYlWQ5cUFX3VlUBO4DrRjJgSdJpLcQ5i3clebAdppp7odIK4Im+NjOttqLNn1wfKMnmJNNJpmdnZ8/0uCVpyRp3WNwGvBxYCxyh91Il6L0j42Q1T32gqtpWVVNVNTUxMfETDlWSNGesYVFVT1bViap6GvgksK6tmgFW9jWdBA63+uSAuiRpjMYaFu0cxJw3AXNXSu0GNiY5N8ll9E5k762qI8BTSa5qV0HdANw5zjFLkp7Fm/KerSSfBV4HXJxkBngf8Loka+kdSnoMeCdAVe1Psovem/iOA1uq6kTb1I30rqw6H7irTZKkMRpZWFTV2waUPzVP+63A1gH1aeCKMzg0SdKz5B3ckqROhoUkqZNhIUnqZFhIkjoZFpKkToaFJKmTYSFJ6mRYSJI6GRaSpE6GhSSpk2EhSepkWEiSOhkWkqROhoUkqZNhIUnqZFhIkjoZFpKkTiMLiyS3Jzma5KG+2kVJ7k7yaPu8sG/dzUkOJnkkyTV99SuT7Gvrbmnv4pYkjdEo9yzuANafVLsJ2FNVq4E9bZkka4CNwOWtz61JlrU+twGbgdVtOnmbkqQRG1lYVNVXge+fVN4AbG/z24Hr+uo7q+pYVR0CDgLrkiwHLqiqe6uqgB19fSRJYzLucxaXVtURgPZ5SauvAJ7oazfTaiva/Mn1gZJsTjKdZHp2dvaMDlySlrLFcoJ70HmImqc+UFVtq6qpqpqamJg4Y4OTpKVu3GHxZDu0RPs82uozwMq+dpPA4VafHFCXJI3RuMNiN7CpzW8C7uyrb0xybpLL6J3I3tsOVT2V5Kp2FdQNfX0kSWNyzqg2nOSzwOuAi5PMAO8DPgjsSvIO4HHgLQBVtT/JLuAAcBzYUlUn2qZupHdl1fnAXW2SJI3RyMKiqt52mlVXn6b9VmDrgPo0cMUZHJok6VlaLCe4JUmLmGEhSepkWEiSOhkWkqROhoUkqZNhIUnqZFhIkjqN7D4LSaPz+G+8aqGHoEXoZf9638i27Z6FJKmTYSFJ6mRYSJI6GRaSpE6GhSSpk2EhSepkWEiSOhkWkqROhoUkqdOChEWSx5LsS/JAkulWuyjJ3UkebZ8X9rW/OcnBJI8kuWYhxixJS9lC7ln8bFWtraqptnwTsKeqVgN72jJJ1gAbgcuB9cCtSZYtxIAlaalaTIehNgDb2/x24Lq++s6qOlZVh4CDwLrxD0+Slq6FCosC/nuS+5NsbrVLq+oIQPu8pNVXAE/09Z1ptVMk2ZxkOsn07OzsiIYuSUvPQj119rVVdTjJJcDdSb49T9sMqNWghlW1DdgGMDU1NbCNJOnZW5A9i6o63D6PAl+gd1jpySTLAdrn0dZ8BljZ130SODy+0UqSxh4WSf5mkpfMzQM/DzwE7AY2tWabgDvb/G5gY5Jzk1wGrAb2jnfUkrS0LcRhqEuBLySZ+/7frarfT/InwK4k7wAeB94CUFX7k+wCDgDHgS1VdWIBxi1JS9bYw6Kq/gx49YD694CrT9NnK7B1xEOTJJ3GYrp0VpK0SBkWkqROhoUkqZNhIUnqZFhIkjoZFpKkToaFJKmTYSFJ6mRYSJI6GRaSpE6GhSSpk2EhSepkWEiSOhkWkqROhoUkqZNhIUnqZFhIkjoZFpKkTmdNWCRZn+SRJAeT3LTQ45GkpeSsCIsky4DfAv4xsAZ4W5I1CzsqSVo6zoqwANYBB6vqz6rq/wE7gQ0LPCZJWjLOWegBDGkF8ETf8gzwD05ulGQzsLkt/jDJI2MY21JwMfDdhR7EYpAPbVroIehU/v2c876cia387UHFsyUsBv0J1CmFqm3AttEPZ2lJMl1VUws9DmkQ/36Ox9lyGGoGWNm3PAkcXqCxSNKSc7aExZ8Aq5NcluRFwEZg9wKPSZKWjLPiMFRVHU/yLuC/AcuA26tq/wIPaynx0J4WM/9+jkGqTjn0L0nSjzlbDkNJkhaQYSFJ6mRYaF4+ZkWLVZLbkxxN8tBCj2UpMCx0Wj5mRYvcHcD6hR7EUmFYaD4+ZkWLVlV9Ffj+Qo9jqTAsNJ9Bj1lZsUBjkbSADAvNZ6jHrEh6/jMsNB8fsyIJMCw0Px+zIgkwLDSPqjoOzD1m5WFgl49Z0WKR5LPAvcArk8wkecdCj+n5zMd9SJI6uWchSepkWEiSOhkWkqROhoUkqZNhIUnqZFhIfZL88KTlX07yiTF+/x1Jfmlc3ycNy7CQFkh7qq90VjAspCEl+YUkX0/yzSRfTnJpq38pyQNt+oskm5KsSvJHSb7Rpp9pbV+X5CtJfhfYl55PJDmQ5L8Cl/R939Xtu/a1dzec2+qPJbm4zU8luafN/6O+cXwzyUvG/Eek57FzFnoA0iJzfpIH+pYv4plHnHwNuKqqKsmvAr8OvLeqrgVIciXwO8DvAX8FvKGqfpRkNfBZYKptZx1wRVUdSvJm4JXAq4BLgQPA7UnOo/e+hqur6k+T7ABuBH5znrH/GrClqv44yYuBHz33PwbpxxkW0o/7y6paO7eQ5Jd55j/5SeA/JlkOvAg41NfuYuDTwFur6i+S/BTwiSRrgRPAK/q+Y29VzfX9h8Bnq+oEcDjJH7T6K4FDVfWnbXk7sIX5w+KPgY8k+Qzw+aqaeTY/XJqPh6Gk4X0c+ERVvQp4J3Ae/PW5h53Ab1TV3Cs+/wXwJPBqemHzor7t/N+TtjvomTuDHg8/5zjP/Ns97683UvVB4FeB84H7kvzdIX6TNBTDQhreTwHfafOb+uofBB6sqp0ntT1SVU8D1wOnO5n9VWBjkmVtj+VnW/3bwKokf6ctXw/8YZt/DLiyzf/i3IaSvLyq9lXVvwWmAcNCZ4xhIQ3v/cB/SvJHwHf76r8G/HzfyeU3ArcCm5LcR+8Q1Ml7E3O+ADwK7ANuowVCVf0I+JX2ffuAp4F/3/p8APhYG8eJvm29J8lDSb4F/CVw10/6g6U5PnVWktTJPQtJUifDQpLUybCQJHUyLCRJnQwLSVInw0KS1MmwkCR1+v8JzBXsIELgqgAAAABJRU5ErkJggg==\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "# checking the distribution of outcomes\n",
    "sns.countplot(x = 'Hazardous', data = df)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Splitting Data for Training and Testing"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "data = df.values\n",
    "X, y = data[:,:-1], data[:,-1]\n",
    "X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20, random_state=0)  # splitting in the ratio 80:20"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Making Predictions using Naive Bayes Model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "gnb = GaussianNB()\n",
    "y_pred = gnb.fit(X_train, y_train).predict(X_test)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Checking Accuracy"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.8454157782515992"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "score = gnb.score(X_test, y_test)\n",
    "score"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Predictions are 84.54% accurate."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Results' Visualization"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([[793,   0],\n",
       "       [145,   0]], dtype=int64)"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cm = confusion_matrix(y_test, y_pred)\n",
    "cm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAXQAAAFaCAYAAAAOxaYVAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjMuMiwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8vihELAAAACXBIWXMAAAsTAAALEwEAmpwYAAAcL0lEQVR4nO3de7RdZX3u8e9DIhclFy6SBhIrShSBU1ERba2WiC0IFqTKMVgdHBuNeqi3Y3sAPcMe0Fg6bGs9VipR1KACDVWEqkPEeMFbBUTuSIkiEBMCgkCUGkzyO3+sGViEvdfeSfZeazH5fsbYY835rnfO90fIeDL3u+Z6Z6oKSdKj33aDLkCSNDEMdElqCQNdklrCQJekljDQJaklDHRJaompgy5AkobB2rVrt+ke7mnTpmWiatlaXqFLUksY6JLUEga6JLWEgS5JLWGgS1JLGOiS1BIGuiS1hIEuSS1hoEtSSxjoktQSBroktYSBLkktYaBLUksY6JLUEga6JLWEgS5JLWGgS1JLGOiS1BIGuiS1hIEuSS1hoEtSSxjoktQSBroktYSBLkktYaBLUksY6JLUEga6JLWEgS5JLTF10AX0UIMuQNKjRgZdwDAY5kBn7dq1gy5BQ2TatGkArP/S+QOuRMNk6pHHDLqEoeGUiyS1hIEuSS1hoEtSSxjoktQHSZ6e5Mqun/uSvD3JrkkuTnJT87pL1zEnJ1mR5MYkh401hoEuSX1QVTdW1YFVdSDwHOB+4HzgJGB5Vc0Dljf7JNkPWADsDxwOnJ5kSq8xhvouF0nqlyfc/sC2nWDaFvU+FPhJVd2S5GjgkKZ9KfBN4ETgaODcqloH3JxkBXAw8P3RTuoVuiT13wLgnGZ7VlWtBmhe92ja9wJu6zpmZdM2KgNdkiZAkkVJLu/6WTRKv+2Bo4DzxjrlCG09v3DplIskTYCqWgIsGUfXlwJXVNWaZn9NktlVtTrJbOCOpn0lMLfruDnAql4n9gpdkvrrOB6abgG4EDi+2T4euKCrfUGSHZLsDcwDLu11Yq/QJalPkjwe+GPgjV3NpwHLkiwEbgWOBaiq65IsA64H1gMnVNWGXuc30CWpT6rqfmC3zdruonPXy0j9FwOLx3t+p1wkqSUMdElqCQNdklrCQJekljDQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6SWMNAlqSUMdElqCQNdklrCQJekljDQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6SWMNAlqSUMdElqCQNdklrCQJekljDQJaklDHRJaompgy5AkobB7Ruv2qbj9+TFE1TJ1vMKXZJawkCXpJYw0CWpT5LMTPJvSX6c5IYkv59k1yQXJ7mped2lq//JSVYkuTHJYWOd30CXpP75EPCVqtoXeCZwA3ASsLyq5gHLm32S7AcsAPYHDgdOTzKl18kNdEnqgyTTgRcBZwJU1QNVdQ9wNLC06bYUeHmzfTRwblWtq6qbgRXAwb3GMNAlaQIkWZTk8q6fRZt1eQpwJ/DJJD9K8vEkTwBmVdVqgOZ1j6b/XsBtXcevbNpG5W2LkjQBqmoJsKRHl6nAs4G3VNUPknyIZnplFBlpmF41eIUuSf2xElhZVT9o9v+NTsCvSTIboHm9o6v/3K7j5wCreg1goEtSH1TV7cBtSZ7eNB0KXA9cCBzftB0PXNBsXwgsSLJDkr2BecClvcZwykWS+uctwGeTbA/8FHgdnQvrZUkWArcCxwJU1XVJltEJ/fXACVW1odfJDXRJ6pOquhI4aIS3Dh2l/2Jg8XjP75SLJLWEgS5JLWGgS1JLGOiS1BIGuiS1hIEuSS3hbYsD9rOf/Yx3vetdD+7//Oc/541vfCMHHXQQf/u3f8v999/PnnvuyXvf+1523nlnrr32Wt7//vcDUFUsWrSI+fPnD6p8Dci3b7iR077w72zYWLzi+c/lDYceMuiSNAQM9AF78pOfzNlnnw3Ahg0bOOKII5g/fz4nnngib3vb23jOc57DBRdcwKc//Wne/OY3s88++3DWWWcxdepUfvGLX3Dcccfxwhe+kKlT/V/5WLFh40YWf/4CPvamhcyaMYNXffCfmb//M9jnd2YNujQNmFMuQ+Syyy5jr732Yvbs2dxyyy08+9nPBuB5z3seX//61wHYcccdHwzvdevWkYy0fo/a7Jpbb2Pu7rsxd7fd2H7qVI541jP5xrXXD7osDYFJu6xLsi+d9Xz3orNC2Crgwqq6YbLGfLS76KKLOOywzkNJnvrUp/Ktb32LQw45hK997WusWbPmwX7XXnstp556KqtXr+bUU0/16vwxZs299zF75owH92fNnMHVt9zW4wg9VkzKFXqSE4Fz6Sz/eClwWbN9TpJey0U+Zv32t7/lkksu4SUveQkA73nPezjvvPN4zWtew/3338/jHve4B/secMABLFu2jLPOOotPfvKTrFu3blBlaxDqkSuo+ouaYPKu0BcC+1fVb7sbk/wjcB1w2kgHNQvCLwI444wzOO644yapvOHz3e9+l3333ZfddtsN6Mytf+QjHwHglltu4Tvf+c4jjtl7773Zaaed+MlPfsJ+++3X13o1OLNmzmD1Pfc+uL/mnnvZY/r0AVakYTFZc+gbgT1HaJ/dvDeiqlpSVQdV1UGLFm3+sI92655uAbj77rsB2LhxI2eeeSaveMUrgM5dMOvXrwdg9erV3HLLLey550h/1GqrA+bO4dY772LlXXfzwPr1fPlHVzH/AP9B1+Rdob8dWJ7kJh56hNKTgH2Av5ykMR+1fvOb33DppZfy7ne/+8G2iy66iPPOOw+A+fPnc9RRRwFw5ZVXsnTpUqZOnUoSTjrpJGbOnDmIsjUgU6dM4d1/dhSLlnyCjRs3cszBB3mHiwBIjTAfNyEnTraj80DTvejMn68ELhtrPd8utXbt2kmpTY9O06ZNA2D9l84fcCUaJlOPPAZGflzbFll149e3KQz3fPqLB/5JxqTdHlFVG4H/mKzzS5IezvvQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6SWcN1VSQL2fPqMsTsNOa/QJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6Q+SfKzJNckuTLJ5U3brkkuTnJT87pLV/+Tk6xIcmOSw0Y/c4eBLkn9Nb+qDqyqg5r9k4DlVTUPWN7sk2Q/YAGwP3A4cHqSKb1ObKBL0mAdDSxttpcCL+9qP7eq1lXVzcAKOs9pHpWBLkkTIMmiJJd3/SwaoVsBX03yw673Z1XVaoDmdY+mfS/gtq5jVzZto3ItF0maAFW1BFgyRrcXVNWqJHsAFyf5cY++GWmYXif3Cl2S+qSqVjWvdwDn05lCWZNkNkDzekfTfSUwt+vwOcCqXuc30CWpD5I8Icm0TdvAnwDXAhcCxzfdjgcuaLYvBBYk2SHJ3sA84NJeYzjlIkn9MQs4Pwl0svfsqvpKksuAZUkWArcCxwJU1XVJlgHXA+uBE6pqQ68BDHRJ6oOq+inwzBHa7wIOHeWYxcDi8Y7hlIsktYSBLkktYaBLUksY6JLUEga6JLWEgS5JLWGgS1JLGOiS1BIGuiS1hIEuSS0x6lf/k6zloaUaNy3jWM12VdX0Sa5NkrQFRg30qprWz0IkSdtmXFMuSf4wyeua7d2bpRwlSUNkzEBP8jfAicDJTdP2wGcmsyhJ0pYbzxX6McBRwK/hwSduOB0jSUNmPIH+QFUVzQekzZM2JElDZjyBvizJGcDMJG8AvgZ8bHLLkiRtqTGfWFRVf5/kj4H7gKcB76mqiye9MknSFhnvI+iuAXaiM+1yzeSVI0naWmMGepLXA+8Bvk7nS0UfTnJqVX1isouTpH6p//zPbTo+T3vOBFWy9cZzhf7XwLOaB5mSZDfge4CBLklDZDwfiq4E1nbtrwVum5xyJElbq9daLv+r2fw58IMkF9CZQz8auLQPtUmStkCvKZdNXx76SfOzyQWTV44kaWv1WpzrlH4WIknaNuO5y+WJwP8G9gd23NReVS+exLokSVtoPB+Kfhb4MbA3cArwM+CySaxJkrQVxhPou1XVmcBvq+pbVfUXwPMnuS5J0hYaz33ov21eVyc5ElgFzJm8kiRJW2M8gf6+JDOAdwIfBqYD75jUqiRJW2w8i3N9sdm8F5g/ueVIkrZWry8WfZiHHhL9CFX11kmpSJK0VXpdoV/etyok6TEiyRQ6+frzqnpZkl2BfwWeTOcuwv9eVb9s+p4MLAQ2AG+tqot6nbvXF4uWTkj1kqRubwNuoPN5JMBJwPKqOi3JSc3+iUn2AxbQ+Q7QnsDXkjytqjaMduLx3LYoSZoASeYARwIf72o+Gth0Ab0UeHlX+7lVta6qbgZWAAf3Or+BLkkTIMmiJJd3/Swaods/0fnm/cautllVtRqged2jad+Lh69su7JpG9V4n1g0ENOmTRu7kx5zph55zKBLkB6hqpYAS0Z7P8nLgDuq6odJDhnHKTPSML0OGOq7XL54xcrJHkKPIi97duf7bOu/dP6AK9EweRT9A/8C4KgkR9BZF2t6ks8Aa5LMrqrVSWYDdzT9VwJzu46fQ+eLnaPyLhdJ6oOqOhk4GaC5Qv+rqnpNkg8AxwOnNa+blii/EDg7yT/S+VB0HmM8i8K7XCRpsE4DliVZCNwKHAtQVdclWQZcD6wHTuh1hwuMf/ncE4H9cPlcSdpmVfVN4JvN9l3AoaP0WwwsHu95x7t87g24fK4kDTWXz5WklnD5XElqCZfPlaSWcPlcSWqJ8dzl8klG+IJRM5cuSRoS45ly+WLX9o7AMYzxbSVJUv+NZ8rlc937Sc4BvjZpFUmStsrWrLY4D3jSRBciSdo245lDX8vD59Bvp/PNUUnSEBnPlItr2ErSo8CYUy5Jlo+nTZI0WL3WQ98ReDywe5JdeGix9el0lnKUJA2RXlMubwTeTie8f8hDgX4f8JHJLUuStKV6rYf+IeBDSd5SVR/uY02S1Hc/u+9F23T83hNUx7YYz22LG5PM3LSTZJck/3PySpIkbY3xBPobquqeTTtV9UvgDZNWkSRpq4wn0LdL8uDTp5NMAbafvJIkSVtjPGu5XETneXcfpfMFozcBX5nUqiRJW2w8gX4isAh4M507Xb4KfGwyi5Ikbbkxp1yqamNVfbSqXllVrwCuo/OgC0nSEBnPFTpJDgSOA14F3Ax8fhJrkiRthV7fFH0asIBOkN8F/CuQqvKpRZI0hHpdof8Y+Dbwp1W1AiCJzxKVpCHVaw79FXSWyv1Gko8lOZSHvv4vSRoyowZ6VZ1fVa8C9gW+CbwDmJXkX5L8SZ/qkySN03jucvl1VX22ql4GzAGuBE6a7MIkSVtmix5BV1V3V9UZVfXiySpIkrR1tuaZopKkIWSgS1JLGOiS1BIGuiS1hIEuSX2QZMcklya5Ksl1SU5p2ndNcnGSm5rXXbqOOTnJiiQ3JjlsrDEMdEnqj3XAi6vqmcCBwOFJnk/nNvDlVTUPWN7sk2Q/Osuv7A8cDpzePI9iVAa6JPVBdfyq2X1c81PA0cDSpn0p8PJm+2jg3KpaV1U3AyuAg3uNYaBL0gRIsijJ5V0/i0boMyXJlcAdwMVV9QNgVlWtBmhe92i67wXc1nX4yqZtVONaPleS1FtVLQGWjNFnA3BgkpnA+UkO6NF9pLWzqtf5vUKXpD6rqnvorJF1OLAmyWyA5vWOpttKYG7XYXOAVb3Oa6BLUh8keWJzZU6SnYCX0Fmm/ELg+Kbb8cAFzfaFwIIkOyTZG5gHXNprDKdcJKk/ZgNLmztVtgOWVdUXk3wfWJZkIXArcCxAVV2XZBlwPbAeOKGZshmVgS5JfVBVVwPPGqH9LuDQUY5ZDCwe7xhOuUhSSxjoktQSBroktYSBLkktYaBLUksY6JLUEga6JLWEgS5JLWGgS1JLGOiS1BJ+9V+SgLuffMs2Hb9376XK+8IrdElqCQNdklrCQJekljDQJaklDHRJagkDXZJawtsWh8C5H/0AN/zoP9h5+kz++gNnPuy9b3xxGV/87Bmccsbn2Xn6DO6+83b+7p2vY489O8+O/d19nsErX/+OQZStAfr2DTdy2hf+nQ0bi1c8/7m84dBDBl2ShoCBPgSe+0eH8YeHHc05p//dw9p/edcd/Oc1P2SX3fd4WPvus/bknact6WeJGiIbNm5k8ecv4GNvWsisGTN41Qf/mfn7P4N9fmfWoEvTgDnlMgSe+ozf4/E7T39E+4Vnnc6fvnoRkP4XpaF1za23MXf33Zi7225sP3UqRzzrmXzj2usHXZaGgFfoQ+ray7/HjF13Z8/ffeoj3rv7ztv5h5PeyI47PZ6Xvup1PGXf3xtAhRqUNffex+yZMx7cnzVzBlffctsAK9Kw6PsVepLX9XvMR5sH1v2G5V/4LIcd+z8e8d70mbvyfz58Nu887QyOeu2b+cyH389v7v91/4vU4FQ9oin+EicGM+VyymhvJFmU5PIkly9Z8tidI75rzarOVfiJi3jfW17NvXffyQff9Sbuu+dupj5ue54wrXN1NvcpT2P3WXty5+qVA65Y/TRr5gxW33Pvg/tr7rmXPaY/cspOjz2TMuWS5OrR3gJG/eSmqpYAm5K8vnjFYzOoZj/pKZxyxuce3H/fW17N2xf/CztPn8Gv7ruHx+88je22m8Jda1Zx5+0r2W3W7AFWq347YO4cbr3zLlbedTd7zJjOl390FR947XGDLktDYLLm0GcBhwG/3Kw9wPcmacxHrU//v/fxkxuu4tdr7+XUE17FYa88nufNP2LEvj+94Wq+ct6n2G7KFLbbbjteufDtI36gqvaaOmUK7/6zo1i05BNs3LiRYw4+yDtcBEBqhPm4bT5pcibwyar6zgjvnV1Vrx7HaR6zV+ga2cuePQeA9V86f8CVaJhMPfIYmIBbwX74i+9tUxg+Z/c/GPgnGZNyhV5VC3u8N54wlyRtIe9Dl6SWMNAlqSUMdElqCQNdklrCQJekljDQJakPksxN8o0kNyS5LsnbmvZdk1yc5KbmdZeuY05OsiLJjUkOG2sMA12S+mM98M6qegbwfOCEJPsBJwHLq2oesLzZp3lvAbA/cDhwepIpvQYw0CWpD6pqdVVd0WyvBW4A9gKOBpY23ZYCL2+2jwbOrap1VXUzsAI4uNcYBrok9VmSJwPPAn4AzKqq1dAJfWDTE232ArrXRV7ZtI3KQJekCdC9Wmzzs2iUfjsDnwPeXlX39TrlCG09lyfwAReSNAE2Wy12REkeRyfMP1tVn2+a1ySZXVWrk8wG7mjaVwJzuw6fA6zqdX6v0CWpD5IEOBO4oar+seutC4Hjm+3jgQu62hck2SHJ3sA84NJeY3iFLkn98QLgtcA1Sa5s2t4FnAYsS7IQuBU4FqCqrkuyDLiezh0yJ1TVhl4DGOiS1AfNcuKjLbF76CjHLAYWj3cMp1wkqSUMdElqCQNdklrCQJekljDQJaklvMtFkoDVtz5p206w+8TUsS28QpekljDQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6SWMNAlqSUMdElqCQNdklrCQJekljDQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWsJAl6SWMNAlqSUMdElqCQNdklrCQJekljDQJaklDHRJ6oMkn0hyR5Jru9p2TXJxkpua11263js5yYokNyY5bDxjGOiS1B+fAg7frO0kYHlVzQOWN/sk2Q9YAOzfHHN6kiljDWCgS1IfVNUlwN2bNR8NLG22lwIv72o/t6rWVdXNwArg4LHGMNAlaXBmVdVqgOZ1j6Z9L+C2rn4rm7aeDHRJmgBJFiW5vOtn0bacboS2GuugqdswoCSpUVVLgCVbeNiaJLOranWS2cAdTftKYG5XvznAqrFO5hW6JA3OhcDxzfbxwAVd7QuS7JBkb2AecOlYJ/MKXZL6IMk5wCHA7klWAn8DnAYsS7IQuBU4FqCqrkuyDLgeWA+cUFUbxhrDQJekPqiq40Z569BR+i8GFm/JGAa6JAGHr75sG88wZ0Lq2BbOoUtSSxjoktQSBroktYSBLkktYaBLUksM9V0uL3v24D811vCZeuQxgy5BGkrDHOgjrWXwmJRkUfO1YulB/r3Q5pxyeXTYlkV+1F7+vdDDGOiS1BIGuiS1hIH+6OA8qUbi3ws9jIH+KOAHXxqJfy+0OQNdklrCQB9ySQ5PcmOSFUlOGnQ9Grwkn0hyR5JrB12LhouBPsSSTAE+ArwU2A84Lsl+g61KQ+BTwOGDLkLDx0AfbgcDK6rqp1X1AHAucPSAa9KAVdUlwN2DrkPDx0AfbnsBt3Xtr2zaJOkRDPThNtLyB9X3KiQ9Khjow20lMLdrfw6wakC1SBpyBvpwuwyYl2TvJNsDC4ALB1yTpCFloA+xqloP/CVwEXADsKyqrhtsVRq0JOcA3weenmRlkoWDrknDYZiXzxVQVV8GvjzoOjQ8quq4Qdeg4eQVuiS1hIEuSS1hoEtSSxjoktQSBroktYSBrp6SbEhyZZJrk5yX5PHbcK5PJXlls/3xXguNJTkkyR9sxRg/S7L7eNs36/OrLRzr/yb5qy2tUZosBrrG8l9VdWBVHQA8ALyp+81mRcgtVlWvr6rre3Q5BNjiQJceywx0bYlvA/s0V8/fSHI2cE2SKUk+kOSyJFcneSNAOv45yfVJvgTsselESb6Z5KBm+/AkVyS5KsnyJE+m8w/HO5rfDl6Y5IlJPteMcVmSFzTH7pbkq0l+lOQMRl7/5mGSfCHJD5Ncl2TRZu/9Q1PL8iRPbNqemuQrzTHfTrLvhPxpShPMLxZpXJJMpbMu+1eapoOBA6rq5iYU762q5ybZAfhukq8CzwKeDvw3YBZwPfCJzc77ROBjwIuac+1aVXcn+Sjwq6r6+6bf2cAHq+o7SZ5E59uzzwD+BvhOVZ2a5EjgYQE9ir9oxtgJuCzJ56rqLuAJwBVV9c4k72nO/Zd0nt35pqq6KcnzgNOBF2/FH6M0qQx0jWWnJFc2298GzqQzFXJpVd3ctP8J8Hub5seBGcA84EXAOVW1AViV5OsjnP/5wCWbzlVVo63z/RJgv+TBC/DpSaY1Y/xZc+yXkvxyHP9Nb01yTLM9t6n1LmAj8K9N+2eAzyfZufnvPa9r7B3GMYb0CEkOBz4ETAE+XlWnTeT5DXSN5b+q6sDuhibYft3dBLylqi7arN8RjL3cb8bRBzrTg79fVf81Qi3jXlI4ySF0/nH4/aq6P8k3gR1H6V7NuPds/mcgbamuJ5D9MZ2VVC9LcuEYnyVtEefQNREuAt6c5HEASZ6W5AnAJcCCZo59NjB/hGO/D/xRkr2bY3dt2tcC07r6fZXO9AdNvwObzUuAP2/aXgrsMkatM4BfNmG+L53fEDbZDtj0W8ar6Uzl3AfcnOTYZowkeeYYY0gjmfQnkBnomggfpzM/fkXz4OIz6Pz2dz5wE3AN8C/AtzY/sKrupDPv/fkkV/HQlMe/A8ds+lAUeCtwUPOh6/U8dLfNKcCLklxBZ+rn1jFq/QowNcnVwHuB/+h679fA/kl+SGeO/NSm/c+BhU191+FjALV1Jv0JZKnyATiStK2amwO6P5RfUlVLut4/Fjisql7f7L8WOLiq3jJRNTiHLkkToAnvJT26TPoTyJxykaT+mPQnkHmFLkl9UFXrk2x6AtkU4BMT/QQy59AlqSWccpGkljDQJaklDHRJagkDXZJawkCXpJYw0CWpJQx0SWoJA12SWuL/A9VJbRZxpkPVAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<Figure size 432x432 with 2 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.figure(figsize=(6,6))\n",
    "sns.heatmap(cm, annot=True, fmt=\".0f\", linewidths=0.5, square = True, cmap = 'Pastel1')\n",
    "plt.ylabel('Actual label')\n",
    "plt.xlabel('Predicted label')\n",
    "plt.show()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}