code/creating_a_wildfire_database_for_the_datathon.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "7b9a56f5",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd \n",
    "import numpy as np \n",
    "import matplotlib.pyplot as plt\n",
    "import seaborn as sns\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "7fd87ae8",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires = pd.read_excel('fp-historical-wildfire-data-2006-2021.xlsx')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "fe7b1590",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 22914 entries, 0 to 22913\n",
      "Data columns (total 50 columns):\n",
      " #   Column                        Non-Null Count  Dtype         \n",
      "---  ------                        --------------  -----         \n",
      " 0   fire_year                     22914 non-null  int64         \n",
      " 1   fire_number                   22914 non-null  object        \n",
      " 2   fire_name                     604 non-null    object        \n",
      " 3   current_size                  22914 non-null  float64       \n",
      " 4   size_class                    22914 non-null  object        \n",
      " 5   fire_location_latitude        22914 non-null  float64       \n",
      " 6   fire_location_longitude       22914 non-null  float64       \n",
      " 7   fire_origin                   22905 non-null  object        \n",
      " 8   general_cause_desc            22914 non-null  object        \n",
      " 9   industry_identifier_desc      408 non-null    object        \n",
      " 10  responsible_group_desc        9031 non-null   object        \n",
      " 11  activity_class                14204 non-null  object        \n",
      " 12  true_cause                    12488 non-null  object        \n",
      " 13  fire_start_date               22294 non-null  object        \n",
      " 14  det_agent                     22913 non-null  object        \n",
      " 15  det_agent_type                22913 non-null  object        \n",
      " 16  discovered_date               18936 non-null  datetime64[ns]\n",
      " 17  discovered_size               0 non-null      float64       \n",
      " 18  reported_date                 22913 non-null  datetime64[ns]\n",
      " 19  dispatched_resource           22908 non-null  object        \n",
      " 20  dispatch_date                 22908 non-null  datetime64[ns]\n",
      " 21  start_for_fire_date           22907 non-null  datetime64[ns]\n",
      " 22  assessment_resource           22914 non-null  object        \n",
      " 23  assessment_datetime           22914 non-null  datetime64[ns]\n",
      " 24  assessment_hectares           22914 non-null  float64       \n",
      " 25  fire_spread_rate              20305 non-null  float64       \n",
      " 26  fire_type                     20437 non-null  object        \n",
      " 27  fire_position_on_slope        20265 non-null  object        \n",
      " 28  weather_conditions_over_fire  20258 non-null  object        \n",
      " 29  temperature                   20257 non-null  float64       \n",
      " 30  relative_humidity             20255 non-null  float64       \n",
      " 31  wind_direction                20254 non-null  object        \n",
      " 32  wind_speed                    20254 non-null  float64       \n",
      " 33  fuel_type                     16026 non-null  object        \n",
      " 34  initial_action_by             22907 non-null  object        \n",
      " 35  ia_arrival_at_fire_date       16387 non-null  datetime64[ns]\n",
      " 36  ia_access                     10076 non-null  object        \n",
      " 37  fire_fighting_start_date      16507 non-null  datetime64[ns]\n",
      " 38  fire_fighting_start_size      16507 non-null  float64       \n",
      " 39  bucketing_on_fire             16338 non-null  object        \n",
      " 40  distance_from_water_source    6365 non-null   float64       \n",
      " 41  first_bucket_drop_date        6365 non-null   datetime64[ns]\n",
      " 42  bh_fs_date                    22914 non-null  datetime64[ns]\n",
      " 43  bh_hectares                   22914 non-null  float64       \n",
      " 44  uc_fs_date                    22914 non-null  datetime64[ns]\n",
      " 45  uc_hectares                   22914 non-null  float64       \n",
      " 46  to_fs_date                    2444 non-null   datetime64[ns]\n",
      " 47  to_hectares                   2444 non-null   float64       \n",
      " 48  ex_fs_date                    22914 non-null  datetime64[ns]\n",
      " 49  ex_hectares                   22914 non-null  float64       \n",
      "dtypes: datetime64[ns](12), float64(15), int64(1), object(22)\n",
      "memory usage: 8.7+ MB\n"
     ]
    }
   ],
   "source": [
    "wildfires.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "1a666afb",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Observe that there are a few columns where the number of non-Null values a very similar: \n",
    "# fire_spread_rate, fire_type, fire_position_on_slope, weather_conditions_over_fire, temperature, \n",
    "# relative humidity, wind_direction, and wind speed\n",
    "# So let's see if those null values are in similar rows in the data set"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "4958181c",
   "metadata": {},
   "outputs": [],
   "source": [
    "columns_with_similar_null_totals = [\n",
    "    'fire_spread_rate',\n",
    "    'fire_type', \n",
    "    'fire_position_on_slope', \n",
    "    'weather_conditions_over_fire', \n",
    "    'temperature',\n",
    "    'relative_humidity',\n",
    "    'wind_direction', \n",
    "    'wind_speed'\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "21503e12",
   "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>fire_year</th>\n",
       "      <th>fire_number</th>\n",
       "      <th>fire_name</th>\n",
       "      <th>current_size</th>\n",
       "      <th>size_class</th>\n",
       "      <th>fire_location_latitude</th>\n",
       "      <th>fire_location_longitude</th>\n",
       "      <th>fire_origin</th>\n",
       "      <th>general_cause_desc</th>\n",
       "      <th>industry_identifier_desc</th>\n",
       "      <th>...</th>\n",
       "      <th>distance_from_water_source</th>\n",
       "      <th>first_bucket_drop_date</th>\n",
       "      <th>bh_fs_date</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_fs_date</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>to_fs_date</th>\n",
       "      <th>to_hectares</th>\n",
       "      <th>ex_fs_date</th>\n",
       "      <th>ex_hectares</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF012</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>49.599117</td>\n",
       "      <td>-114.353217</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-04-17 14:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-04-17 14:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-04-17 14:36:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF025</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>51.400830</td>\n",
       "      <td>-115.076775</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-05-21 16:00:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-05-21 16:00:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-05-21 16:15:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF026</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>51.391783</td>\n",
       "      <td>-115.109717</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-05-22 16:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-05-22 16:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-05-22 16:50:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF027</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>51.391797</td>\n",
       "      <td>-115.107490</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-05-22 16:57:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-05-22 16:57:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-05-22 17:05:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF028</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>51.602402</td>\n",
       "      <td>-115.013847</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-05-22 17:06:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-05-22 17:06:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-05-22 17:10:00</td>\n",
       "      <td>0.01</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>15320</th>\n",
       "      <td>2010</td>\n",
       "      <td>LWF117</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>54.545083</td>\n",
       "      <td>-111.475700</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2010-05-24 16:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2010-05-24 16:35:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2010-05-24 18:52:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15739</th>\n",
       "      <td>2010</td>\n",
       "      <td>RWF048</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>52.229657</td>\n",
       "      <td>-116.449037</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2010-06-28 15:41:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2010-06-28 15:41:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2010-06-28 15:46:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15748</th>\n",
       "      <td>2010</td>\n",
       "      <td>RWF057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>52.277024</td>\n",
       "      <td>-116.038532</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2010-07-15 14:10:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2010-07-15 14:10:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2010-07-15 14:30:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15750</th>\n",
       "      <td>2010</td>\n",
       "      <td>RWF059</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>51.858517</td>\n",
       "      <td>-115.101837</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2010-07-25 12:00:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2010-07-25 12:00:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2010-07-25 12:01:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22219</th>\n",
       "      <td>2006</td>\n",
       "      <td>MWF174</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>56.307583</td>\n",
       "      <td>-110.914333</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2006-12-29 11:41:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2006-12-29 11:42:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2007-01-08 14:50:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>2458 rows × 50 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       fire_year fire_number fire_name  current_size size_class  \\\n",
       "11          2021      CWF012       NaN          0.01          A   \n",
       "24          2021      CWF025       NaN          0.01          A   \n",
       "25          2021      CWF026       NaN          0.01          A   \n",
       "26          2021      CWF027       NaN          0.01          A   \n",
       "27          2021      CWF028       NaN          0.01          A   \n",
       "...          ...         ...       ...           ...        ...   \n",
       "15320       2010      LWF117       NaN          0.01          A   \n",
       "15739       2010      RWF048       NaN          0.01          A   \n",
       "15748       2010      RWF057       NaN          0.01          A   \n",
       "15750       2010      RWF059       NaN          0.01          A   \n",
       "22219       2006      MWF174       NaN          0.01          A   \n",
       "\n",
       "       fire_location_latitude  fire_location_longitude      fire_origin  \\\n",
       "11                  49.599117              -114.353217  Provincial Land   \n",
       "24                  51.400830              -115.076775  Provincial Land   \n",
       "25                  51.391783              -115.109717  Provincial Land   \n",
       "26                  51.391797              -115.107490  Provincial Land   \n",
       "27                  51.602402              -115.013847  Provincial Land   \n",
       "...                       ...                      ...              ...   \n",
       "15320               54.545083              -111.475700  Provincial Land   \n",
       "15739               52.229657              -116.449037  Provincial Land   \n",
       "15748               52.277024              -116.038532  Provincial Land   \n",
       "15750               51.858517              -115.101837  Provincial Land   \n",
       "22219               56.307583              -110.914333  Provincial Land   \n",
       "\n",
       "      general_cause_desc industry_identifier_desc  ...  \\\n",
       "11            Recreation                      NaN  ...   \n",
       "24            Recreation                      NaN  ...   \n",
       "25            Recreation                      NaN  ...   \n",
       "26            Recreation                      NaN  ...   \n",
       "27            Recreation                      NaN  ...   \n",
       "...                  ...                      ...  ...   \n",
       "15320         Recreation                      NaN  ...   \n",
       "15739         Recreation                      NaN  ...   \n",
       "15748         Recreation                      NaN  ...   \n",
       "15750         Recreation                      NaN  ...   \n",
       "22219         Recreation                      NaN  ...   \n",
       "\n",
       "      distance_from_water_source first_bucket_drop_date          bh_fs_date  \\\n",
       "11                           NaN                    NaT 2021-04-17 14:35:00   \n",
       "24                           NaN                    NaT 2021-05-21 16:00:00   \n",
       "25                           NaN                    NaT 2021-05-22 16:35:00   \n",
       "26                           NaN                    NaT 2021-05-22 16:57:00   \n",
       "27                           NaN                    NaT 2021-05-22 17:06:00   \n",
       "...                          ...                    ...                 ...   \n",
       "15320                        NaN                    NaT 2010-05-24 16:35:00   \n",
       "15739                        NaN                    NaT 2010-06-28 15:41:00   \n",
       "15748                        NaN                    NaT 2010-07-15 14:10:00   \n",
       "15750                        NaN                    NaT 2010-07-25 12:00:00   \n",
       "22219                        NaN                    NaT 2006-12-29 11:41:00   \n",
       "\n",
       "      bh_hectares          uc_fs_date uc_hectares to_fs_date  to_hectares  \\\n",
       "11           0.01 2021-04-17 14:35:00        0.01        NaT          NaN   \n",
       "24           0.01 2021-05-21 16:00:00        0.01        NaT          NaN   \n",
       "25           0.01 2021-05-22 16:35:00        0.01        NaT          NaN   \n",
       "26           0.01 2021-05-22 16:57:00        0.01        NaT          NaN   \n",
       "27           0.01 2021-05-22 17:06:00        0.01        NaT          NaN   \n",
       "...           ...                 ...         ...        ...          ...   \n",
       "15320        0.01 2010-05-24 16:35:00        0.01        NaT          NaN   \n",
       "15739        0.01 2010-06-28 15:41:00        0.01        NaT          NaN   \n",
       "15748        0.01 2010-07-15 14:10:00        0.01        NaT          NaN   \n",
       "15750        0.01 2010-07-25 12:00:00        0.01        NaT          NaN   \n",
       "22219        0.01 2006-12-29 11:42:00        0.01        NaT          NaN   \n",
       "\n",
       "               ex_fs_date ex_hectares  \n",
       "11    2021-04-17 14:36:00        0.01  \n",
       "24    2021-05-21 16:15:00        0.01  \n",
       "25    2021-05-22 16:50:00        0.01  \n",
       "26    2021-05-22 17:05:00        0.01  \n",
       "27    2021-05-22 17:10:00        0.01  \n",
       "...                   ...         ...  \n",
       "15320 2010-05-24 18:52:00        0.01  \n",
       "15739 2010-06-28 15:46:00        0.01  \n",
       "15748 2010-07-15 14:30:00        0.01  \n",
       "15750 2010-07-25 12:01:00        0.01  \n",
       "22219 2007-01-08 14:50:00        0.01  \n",
       "\n",
       "[2458 rows x 50 columns]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "wildfires[wildfires[columns_with_similar_null_totals].isnull().all(axis = 1)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "2a0be33a",
   "metadata": {},
   "outputs": [],
   "source": [
    "#So there are 2458 rows where every one of those values is null, so I will remove all of those rows"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "214d4f1d",
   "metadata": {},
   "outputs": [],
   "source": [
    "null_row_indices = wildfires[wildfires[columns_with_similar_null_totals].isnull().all(axis = 1)].index.tolist()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "26a00199",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires = wildfires.drop(index = null_row_indices).reset_index(drop=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "5df68a3c",
   "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>fire_year</th>\n",
       "      <th>fire_number</th>\n",
       "      <th>fire_name</th>\n",
       "      <th>current_size</th>\n",
       "      <th>size_class</th>\n",
       "      <th>fire_location_latitude</th>\n",
       "      <th>fire_location_longitude</th>\n",
       "      <th>fire_origin</th>\n",
       "      <th>general_cause_desc</th>\n",
       "      <th>industry_identifier_desc</th>\n",
       "      <th>...</th>\n",
       "      <th>distance_from_water_source</th>\n",
       "      <th>first_bucket_drop_date</th>\n",
       "      <th>bh_fs_date</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_fs_date</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>to_fs_date</th>\n",
       "      <th>to_hectares</th>\n",
       "      <th>ex_fs_date</th>\n",
       "      <th>ex_hectares</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF001</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>49.410191</td>\n",
       "      <td>-114.318843</td>\n",
       "      <td>Provincial Park</td>\n",
       "      <td>Power Line Industry</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-01-13 13:25:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-01-13 13:25:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-01-13 13:35:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF002</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>49.587753</td>\n",
       "      <td>-114.378655</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-01-29 10:45:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2021-01-29 10:45:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-01-29 13:15:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF003</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.05</td>\n",
       "      <td>A</td>\n",
       "      <td>51.119500</td>\n",
       "      <td>-114.778950</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Resident</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-02-23 10:42:00</td>\n",
       "      <td>0.05</td>\n",
       "      <td>2021-02-23 10:42:00</td>\n",
       "      <td>0.05</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-02-23 10:52:00</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF004</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.15</td>\n",
       "      <td>B</td>\n",
       "      <td>51.271888</td>\n",
       "      <td>-115.025818</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Incendiary</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-03-14 19:00:00</td>\n",
       "      <td>0.15</td>\n",
       "      <td>2021-03-14 19:00:00</td>\n",
       "      <td>0.15</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-03-15 11:05:00</td>\n",
       "      <td>0.15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2021</td>\n",
       "      <td>CWF005</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.30</td>\n",
       "      <td>B</td>\n",
       "      <td>51.161000</td>\n",
       "      <td>-114.963283</td>\n",
       "      <td>Indian Reservation</td>\n",
       "      <td>Resident</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2021-03-17 15:01:00</td>\n",
       "      <td>0.80</td>\n",
       "      <td>2021-03-17 16:10:00</td>\n",
       "      <td>0.80</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2021-03-18 13:00:00</td>\n",
       "      <td>0.30</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>20451</th>\n",
       "      <td>2006</td>\n",
       "      <td>WWF110</td>\n",
       "      <td>NaN</td>\n",
       "      <td>9.71</td>\n",
       "      <td>C</td>\n",
       "      <td>54.697817</td>\n",
       "      <td>-116.218083</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Undetermined</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>5.0</td>\n",
       "      <td>2006-10-03 16:23:00</td>\n",
       "      <td>2006-10-03 18:11:00</td>\n",
       "      <td>5.00</td>\n",
       "      <td>2006-10-03 19:13:00</td>\n",
       "      <td>9.71</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2006-10-12 13:22:00</td>\n",
       "      <td>9.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20452</th>\n",
       "      <td>2006</td>\n",
       "      <td>WWF111</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.80</td>\n",
       "      <td>B</td>\n",
       "      <td>54.068434</td>\n",
       "      <td>-115.668980</td>\n",
       "      <td>Private Land</td>\n",
       "      <td>Recreation</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2006-10-04 16:49:00</td>\n",
       "      <td>1.00</td>\n",
       "      <td>2006-10-04 16:49:00</td>\n",
       "      <td>1.00</td>\n",
       "      <td>2006-10-04 17:35:00</td>\n",
       "      <td>0.8</td>\n",
       "      <td>2006-10-05 11:00:00</td>\n",
       "      <td>0.80</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20453</th>\n",
       "      <td>2006</td>\n",
       "      <td>WWF112</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.10</td>\n",
       "      <td>A</td>\n",
       "      <td>54.651933</td>\n",
       "      <td>-115.460117</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Power Line Industry</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2006-10-05 15:19:00</td>\n",
       "      <td>0.10</td>\n",
       "      <td>2006-10-05 15:19:00</td>\n",
       "      <td>0.10</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2006-10-05 15:25:00</td>\n",
       "      <td>0.10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20454</th>\n",
       "      <td>2006</td>\n",
       "      <td>WWF113</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.05</td>\n",
       "      <td>A</td>\n",
       "      <td>54.619465</td>\n",
       "      <td>-115.516485</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Power Line Industry</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2006-10-27 10:59:00</td>\n",
       "      <td>0.05</td>\n",
       "      <td>2006-10-27 10:59:00</td>\n",
       "      <td>0.05</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2006-10-27 11:05:00</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20455</th>\n",
       "      <td>2006</td>\n",
       "      <td>WWF114</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "      <td>54.492152</td>\n",
       "      <td>-115.352502</td>\n",
       "      <td>Provincial Land</td>\n",
       "      <td>Forest Industry</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaT</td>\n",
       "      <td>2006-12-20 17:10:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>2006-12-20 17:10:00</td>\n",
       "      <td>0.01</td>\n",
       "      <td>NaT</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2007-01-02 15:00:00</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>20456 rows × 50 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       fire_year fire_number fire_name  current_size size_class  \\\n",
       "0           2021      CWF001       NaN          0.01          A   \n",
       "1           2021      CWF002       NaN          0.01          A   \n",
       "2           2021      CWF003       NaN          0.05          A   \n",
       "3           2021      CWF004       NaN          0.15          B   \n",
       "4           2021      CWF005       NaN          0.30          B   \n",
       "...          ...         ...       ...           ...        ...   \n",
       "20451       2006      WWF110       NaN          9.71          C   \n",
       "20452       2006      WWF111       NaN          0.80          B   \n",
       "20453       2006      WWF112       NaN          0.10          A   \n",
       "20454       2006      WWF113       NaN          0.05          A   \n",
       "20455       2006      WWF114       NaN          0.01          A   \n",
       "\n",
       "       fire_location_latitude  fire_location_longitude         fire_origin  \\\n",
       "0                   49.410191              -114.318843     Provincial Park   \n",
       "1                   49.587753              -114.378655     Provincial Land   \n",
       "2                   51.119500              -114.778950     Provincial Land   \n",
       "3                   51.271888              -115.025818     Provincial Land   \n",
       "4                   51.161000              -114.963283  Indian Reservation   \n",
       "...                       ...                      ...                 ...   \n",
       "20451               54.697817              -116.218083     Provincial Land   \n",
       "20452               54.068434              -115.668980        Private Land   \n",
       "20453               54.651933              -115.460117     Provincial Land   \n",
       "20454               54.619465              -115.516485     Provincial Land   \n",
       "20455               54.492152              -115.352502     Provincial Land   \n",
       "\n",
       "        general_cause_desc industry_identifier_desc  ...  \\\n",
       "0      Power Line Industry                      NaN  ...   \n",
       "1               Recreation                      NaN  ...   \n",
       "2                 Resident                      NaN  ...   \n",
       "3               Incendiary                      NaN  ...   \n",
       "4                 Resident                      NaN  ...   \n",
       "...                    ...                      ...  ...   \n",
       "20451         Undetermined                      NaN  ...   \n",
       "20452           Recreation                      NaN  ...   \n",
       "20453  Power Line Industry                      NaN  ...   \n",
       "20454  Power Line Industry                      NaN  ...   \n",
       "20455      Forest Industry                      NaN  ...   \n",
       "\n",
       "      distance_from_water_source first_bucket_drop_date          bh_fs_date  \\\n",
       "0                            NaN                    NaT 2021-01-13 13:25:00   \n",
       "1                            NaN                    NaT 2021-01-29 10:45:00   \n",
       "2                            NaN                    NaT 2021-02-23 10:42:00   \n",
       "3                            NaN                    NaT 2021-03-14 19:00:00   \n",
       "4                            NaN                    NaT 2021-03-17 15:01:00   \n",
       "...                          ...                    ...                 ...   \n",
       "20451                        5.0    2006-10-03 16:23:00 2006-10-03 18:11:00   \n",
       "20452                        NaN                    NaT 2006-10-04 16:49:00   \n",
       "20453                        NaN                    NaT 2006-10-05 15:19:00   \n",
       "20454                        NaN                    NaT 2006-10-27 10:59:00   \n",
       "20455                        NaN                    NaT 2006-12-20 17:10:00   \n",
       "\n",
       "      bh_hectares          uc_fs_date uc_hectares          to_fs_date  \\\n",
       "0            0.01 2021-01-13 13:25:00        0.01                 NaT   \n",
       "1            0.01 2021-01-29 10:45:00        0.01                 NaT   \n",
       "2            0.05 2021-02-23 10:42:00        0.05                 NaT   \n",
       "3            0.15 2021-03-14 19:00:00        0.15                 NaT   \n",
       "4            0.80 2021-03-17 16:10:00        0.80                 NaT   \n",
       "...           ...                 ...         ...                 ...   \n",
       "20451        5.00 2006-10-03 19:13:00        9.71                 NaT   \n",
       "20452        1.00 2006-10-04 16:49:00        1.00 2006-10-04 17:35:00   \n",
       "20453        0.10 2006-10-05 15:19:00        0.10                 NaT   \n",
       "20454        0.05 2006-10-27 10:59:00        0.05                 NaT   \n",
       "20455        0.01 2006-12-20 17:10:00        0.01                 NaT   \n",
       "\n",
       "       to_hectares          ex_fs_date ex_hectares  \n",
       "0              NaN 2021-01-13 13:35:00        0.01  \n",
       "1              NaN 2021-01-29 13:15:00        0.01  \n",
       "2              NaN 2021-02-23 10:52:00        0.05  \n",
       "3              NaN 2021-03-15 11:05:00        0.15  \n",
       "4              NaN 2021-03-18 13:00:00        0.30  \n",
       "...            ...                 ...         ...  \n",
       "20451          NaN 2006-10-12 13:22:00        9.71  \n",
       "20452          0.8 2006-10-05 11:00:00        0.80  \n",
       "20453          NaN 2006-10-05 15:25:00        0.10  \n",
       "20454          NaN 2006-10-27 11:05:00        0.05  \n",
       "20455          NaN 2007-01-02 15:00:00        0.01  \n",
       "\n",
       "[20456 rows x 50 columns]"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "wildfires"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "0da36580",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 20456 entries, 0 to 20455\n",
      "Data columns (total 50 columns):\n",
      " #   Column                        Non-Null Count  Dtype         \n",
      "---  ------                        --------------  -----         \n",
      " 0   fire_year                     20456 non-null  int64         \n",
      " 1   fire_number                   20456 non-null  object        \n",
      " 2   fire_name                     591 non-null    object        \n",
      " 3   current_size                  20456 non-null  float64       \n",
      " 4   size_class                    20456 non-null  object        \n",
      " 5   fire_location_latitude        20456 non-null  float64       \n",
      " 6   fire_location_longitude       20456 non-null  float64       \n",
      " 7   fire_origin                   20447 non-null  object        \n",
      " 8   general_cause_desc            20456 non-null  object        \n",
      " 9   industry_identifier_desc      408 non-null    object        \n",
      " 10  responsible_group_desc        6577 non-null   object        \n",
      " 11  activity_class                11747 non-null  object        \n",
      " 12  true_cause                    10031 non-null  object        \n",
      " 13  fire_start_date               20300 non-null  object        \n",
      " 14  det_agent                     20455 non-null  object        \n",
      " 15  det_agent_type                20455 non-null  object        \n",
      " 16  discovered_date               16548 non-null  datetime64[ns]\n",
      " 17  discovered_size               0 non-null      float64       \n",
      " 18  reported_date                 20455 non-null  datetime64[ns]\n",
      " 19  dispatched_resource           20450 non-null  object        \n",
      " 20  dispatch_date                 20450 non-null  datetime64[ns]\n",
      " 21  start_for_fire_date           20449 non-null  datetime64[ns]\n",
      " 22  assessment_resource           20456 non-null  object        \n",
      " 23  assessment_datetime           20456 non-null  datetime64[ns]\n",
      " 24  assessment_hectares           20456 non-null  float64       \n",
      " 25  fire_spread_rate              20305 non-null  float64       \n",
      " 26  fire_type                     20437 non-null  object        \n",
      " 27  fire_position_on_slope        20265 non-null  object        \n",
      " 28  weather_conditions_over_fire  20258 non-null  object        \n",
      " 29  temperature                   20257 non-null  float64       \n",
      " 30  relative_humidity             20255 non-null  float64       \n",
      " 31  wind_direction                20254 non-null  object        \n",
      " 32  wind_speed                    20254 non-null  float64       \n",
      " 33  fuel_type                     16026 non-null  object        \n",
      " 34  initial_action_by             20449 non-null  object        \n",
      " 35  ia_arrival_at_fire_date       13979 non-null  datetime64[ns]\n",
      " 36  ia_access                     9576 non-null   object        \n",
      " 37  fire_fighting_start_date      14079 non-null  datetime64[ns]\n",
      " 38  fire_fighting_start_size      14079 non-null  float64       \n",
      " 39  bucketing_on_fire             13927 non-null  object        \n",
      " 40  distance_from_water_source    6365 non-null   float64       \n",
      " 41  first_bucket_drop_date        6365 non-null   datetime64[ns]\n",
      " 42  bh_fs_date                    20456 non-null  datetime64[ns]\n",
      " 43  bh_hectares                   20456 non-null  float64       \n",
      " 44  uc_fs_date                    20456 non-null  datetime64[ns]\n",
      " 45  uc_hectares                   20456 non-null  float64       \n",
      " 46  to_fs_date                    2444 non-null   datetime64[ns]\n",
      " 47  to_hectares                   2444 non-null   float64       \n",
      " 48  ex_fs_date                    20456 non-null  datetime64[ns]\n",
      " 49  ex_hectares                   20456 non-null  float64       \n",
      "dtypes: datetime64[ns](12), float64(15), int64(1), object(22)\n",
      "memory usage: 7.8+ MB\n"
     ]
    }
   ],
   "source": [
    "wildfires.info()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "c00fe8ec",
   "metadata": {},
   "source": [
    "### Now we have to determine which columns predict the size_class too well "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "6b1a6ba4",
   "metadata": {},
   "outputs": [],
   "source": [
    "# I suspect current_size, bh_hectares, uc_hectares, and ex_hectares might be the culprits "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "16a0d10c",
   "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>fire_year</th>\n",
       "      <th>current_size</th>\n",
       "      <th>fire_location_latitude</th>\n",
       "      <th>fire_location_longitude</th>\n",
       "      <th>discovered_size</th>\n",
       "      <th>assessment_hectares</th>\n",
       "      <th>fire_spread_rate</th>\n",
       "      <th>temperature</th>\n",
       "      <th>relative_humidity</th>\n",
       "      <th>wind_speed</th>\n",
       "      <th>fire_fighting_start_size</th>\n",
       "      <th>distance_from_water_source</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>to_hectares</th>\n",
       "      <th>ex_hectares</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>fire_year</th>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.008835</td>\n",
       "      <td>0.079834</td>\n",
       "      <td>0.044130</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.000660</td>\n",
       "      <td>0.053430</td>\n",
       "      <td>0.023997</td>\n",
       "      <td>-0.017939</td>\n",
       "      <td>0.019096</td>\n",
       "      <td>0.024184</td>\n",
       "      <td>-0.006796</td>\n",
       "      <td>0.005634</td>\n",
       "      <td>0.008500</td>\n",
       "      <td>0.021763</td>\n",
       "      <td>0.008835</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>current_size</th>\n",
       "      <td>0.008835</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.024976</td>\n",
       "      <td>0.014599</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.037687</td>\n",
       "      <td>0.071773</td>\n",
       "      <td>0.018138</td>\n",
       "      <td>-0.018491</td>\n",
       "      <td>0.050302</td>\n",
       "      <td>0.066583</td>\n",
       "      <td>0.002197</td>\n",
       "      <td>0.901745</td>\n",
       "      <td>0.997393</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>fire_location_latitude</th>\n",
       "      <td>0.079834</td>\n",
       "      <td>0.024976</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>-0.060381</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.037698</td>\n",
       "      <td>0.123244</td>\n",
       "      <td>0.151646</td>\n",
       "      <td>0.030022</td>\n",
       "      <td>0.015118</td>\n",
       "      <td>0.049452</td>\n",
       "      <td>-0.032771</td>\n",
       "      <td>0.014228</td>\n",
       "      <td>0.022466</td>\n",
       "      <td>0.035165</td>\n",
       "      <td>0.024976</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>fire_location_longitude</th>\n",
       "      <td>0.044130</td>\n",
       "      <td>0.014599</td>\n",
       "      <td>-0.060381</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.006420</td>\n",
       "      <td>0.052016</td>\n",
       "      <td>0.024582</td>\n",
       "      <td>-0.040349</td>\n",
       "      <td>0.040629</td>\n",
       "      <td>0.014862</td>\n",
       "      <td>-0.003955</td>\n",
       "      <td>0.017869</td>\n",
       "      <td>0.014860</td>\n",
       "      <td>-0.033819</td>\n",
       "      <td>0.014599</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>discovered_size</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>assessment_hectares</th>\n",
       "      <td>-0.000660</td>\n",
       "      <td>0.037687</td>\n",
       "      <td>0.037698</td>\n",
       "      <td>0.006420</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.127932</td>\n",
       "      <td>0.019437</td>\n",
       "      <td>-0.010928</td>\n",
       "      <td>0.025223</td>\n",
       "      <td>0.447678</td>\n",
       "      <td>-0.001790</td>\n",
       "      <td>0.028499</td>\n",
       "      <td>0.032869</td>\n",
       "      <td>0.578828</td>\n",
       "      <td>0.037687</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>fire_spread_rate</th>\n",
       "      <td>0.053430</td>\n",
       "      <td>0.071773</td>\n",
       "      <td>0.123244</td>\n",
       "      <td>0.052016</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.127932</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.152724</td>\n",
       "      <td>-0.111232</td>\n",
       "      <td>0.134798</td>\n",
       "      <td>0.120372</td>\n",
       "      <td>-0.009101</td>\n",
       "      <td>0.069165</td>\n",
       "      <td>0.066523</td>\n",
       "      <td>0.176180</td>\n",
       "      <td>0.071773</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>temperature</th>\n",
       "      <td>0.023997</td>\n",
       "      <td>0.018138</td>\n",
       "      <td>0.151646</td>\n",
       "      <td>0.024582</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.019437</td>\n",
       "      <td>0.152724</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>-0.274911</td>\n",
       "      <td>-0.017498</td>\n",
       "      <td>0.025513</td>\n",
       "      <td>-0.027851</td>\n",
       "      <td>0.017865</td>\n",
       "      <td>0.017076</td>\n",
       "      <td>0.024049</td>\n",
       "      <td>0.018138</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>relative_humidity</th>\n",
       "      <td>-0.017939</td>\n",
       "      <td>-0.018491</td>\n",
       "      <td>0.030022</td>\n",
       "      <td>-0.040349</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.010928</td>\n",
       "      <td>-0.111232</td>\n",
       "      <td>-0.274911</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>-0.168359</td>\n",
       "      <td>-0.023402</td>\n",
       "      <td>0.019332</td>\n",
       "      <td>-0.020308</td>\n",
       "      <td>-0.018370</td>\n",
       "      <td>-0.080456</td>\n",
       "      <td>-0.018491</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>wind_speed</th>\n",
       "      <td>0.019096</td>\n",
       "      <td>0.050302</td>\n",
       "      <td>0.015118</td>\n",
       "      <td>0.040629</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.025223</td>\n",
       "      <td>0.134798</td>\n",
       "      <td>-0.017498</td>\n",
       "      <td>-0.168359</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.040925</td>\n",
       "      <td>-0.023366</td>\n",
       "      <td>0.039520</td>\n",
       "      <td>0.048538</td>\n",
       "      <td>0.122354</td>\n",
       "      <td>0.050302</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>fire_fighting_start_size</th>\n",
       "      <td>0.024184</td>\n",
       "      <td>0.066583</td>\n",
       "      <td>0.049452</td>\n",
       "      <td>0.014862</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.447678</td>\n",
       "      <td>0.120372</td>\n",
       "      <td>0.025513</td>\n",
       "      <td>-0.023402</td>\n",
       "      <td>0.040925</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.001669</td>\n",
       "      <td>0.029727</td>\n",
       "      <td>0.061607</td>\n",
       "      <td>0.729295</td>\n",
       "      <td>0.066583</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>distance_from_water_source</th>\n",
       "      <td>-0.006796</td>\n",
       "      <td>0.002197</td>\n",
       "      <td>-0.032771</td>\n",
       "      <td>-0.003955</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-0.001790</td>\n",
       "      <td>-0.009101</td>\n",
       "      <td>-0.027851</td>\n",
       "      <td>0.019332</td>\n",
       "      <td>-0.023366</td>\n",
       "      <td>0.001669</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.002453</td>\n",
       "      <td>0.002454</td>\n",
       "      <td>-0.029221</td>\n",
       "      <td>0.002197</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>bh_hectares</th>\n",
       "      <td>0.005634</td>\n",
       "      <td>0.901745</td>\n",
       "      <td>0.014228</td>\n",
       "      <td>0.017869</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.028499</td>\n",
       "      <td>0.069165</td>\n",
       "      <td>0.017865</td>\n",
       "      <td>-0.020308</td>\n",
       "      <td>0.039520</td>\n",
       "      <td>0.029727</td>\n",
       "      <td>0.002453</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.924775</td>\n",
       "      <td>0.663345</td>\n",
       "      <td>0.901745</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>uc_hectares</th>\n",
       "      <td>0.008500</td>\n",
       "      <td>0.997393</td>\n",
       "      <td>0.022466</td>\n",
       "      <td>0.014860</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.032869</td>\n",
       "      <td>0.066523</td>\n",
       "      <td>0.017076</td>\n",
       "      <td>-0.018370</td>\n",
       "      <td>0.048538</td>\n",
       "      <td>0.061607</td>\n",
       "      <td>0.002454</td>\n",
       "      <td>0.924775</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.802835</td>\n",
       "      <td>0.997393</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>to_hectares</th>\n",
       "      <td>0.021763</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.035165</td>\n",
       "      <td>-0.033819</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.578828</td>\n",
       "      <td>0.176180</td>\n",
       "      <td>0.024049</td>\n",
       "      <td>-0.080456</td>\n",
       "      <td>0.122354</td>\n",
       "      <td>0.729295</td>\n",
       "      <td>-0.029221</td>\n",
       "      <td>0.663345</td>\n",
       "      <td>0.802835</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ex_hectares</th>\n",
       "      <td>0.008835</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.024976</td>\n",
       "      <td>0.014599</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.037687</td>\n",
       "      <td>0.071773</td>\n",
       "      <td>0.018138</td>\n",
       "      <td>-0.018491</td>\n",
       "      <td>0.050302</td>\n",
       "      <td>0.066583</td>\n",
       "      <td>0.002197</td>\n",
       "      <td>0.901745</td>\n",
       "      <td>0.997393</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                            fire_year  current_size  fire_location_latitude  \\\n",
       "fire_year                    1.000000      0.008835                0.079834   \n",
       "current_size                 0.008835      1.000000                0.024976   \n",
       "fire_location_latitude       0.079834      0.024976                1.000000   \n",
       "fire_location_longitude      0.044130      0.014599               -0.060381   \n",
       "discovered_size                   NaN           NaN                     NaN   \n",
       "assessment_hectares         -0.000660      0.037687                0.037698   \n",
       "fire_spread_rate             0.053430      0.071773                0.123244   \n",
       "temperature                  0.023997      0.018138                0.151646   \n",
       "relative_humidity           -0.017939     -0.018491                0.030022   \n",
       "wind_speed                   0.019096      0.050302                0.015118   \n",
       "fire_fighting_start_size     0.024184      0.066583                0.049452   \n",
       "distance_from_water_source  -0.006796      0.002197               -0.032771   \n",
       "bh_hectares                  0.005634      0.901745                0.014228   \n",
       "uc_hectares                  0.008500      0.997393                0.022466   \n",
       "to_hectares                  0.021763      1.000000                0.035165   \n",
       "ex_hectares                  0.008835      1.000000                0.024976   \n",
       "\n",
       "                            fire_location_longitude  discovered_size  \\\n",
       "fire_year                                  0.044130              NaN   \n",
       "current_size                               0.014599              NaN   \n",
       "fire_location_latitude                    -0.060381              NaN   \n",
       "fire_location_longitude                    1.000000              NaN   \n",
       "discovered_size                                 NaN              NaN   \n",
       "assessment_hectares                        0.006420              NaN   \n",
       "fire_spread_rate                           0.052016              NaN   \n",
       "temperature                                0.024582              NaN   \n",
       "relative_humidity                         -0.040349              NaN   \n",
       "wind_speed                                 0.040629              NaN   \n",
       "fire_fighting_start_size                   0.014862              NaN   \n",
       "distance_from_water_source                -0.003955              NaN   \n",
       "bh_hectares                                0.017869              NaN   \n",
       "uc_hectares                                0.014860              NaN   \n",
       "to_hectares                               -0.033819              NaN   \n",
       "ex_hectares                                0.014599              NaN   \n",
       "\n",
       "                            assessment_hectares  fire_spread_rate  \\\n",
       "fire_year                             -0.000660          0.053430   \n",
       "current_size                           0.037687          0.071773   \n",
       "fire_location_latitude                 0.037698          0.123244   \n",
       "fire_location_longitude                0.006420          0.052016   \n",
       "discovered_size                             NaN               NaN   \n",
       "assessment_hectares                    1.000000          0.127932   \n",
       "fire_spread_rate                       0.127932          1.000000   \n",
       "temperature                            0.019437          0.152724   \n",
       "relative_humidity                     -0.010928         -0.111232   \n",
       "wind_speed                             0.025223          0.134798   \n",
       "fire_fighting_start_size               0.447678          0.120372   \n",
       "distance_from_water_source            -0.001790         -0.009101   \n",
       "bh_hectares                            0.028499          0.069165   \n",
       "uc_hectares                            0.032869          0.066523   \n",
       "to_hectares                            0.578828          0.176180   \n",
       "ex_hectares                            0.037687          0.071773   \n",
       "\n",
       "                            temperature  relative_humidity  wind_speed  \\\n",
       "fire_year                      0.023997          -0.017939    0.019096   \n",
       "current_size                   0.018138          -0.018491    0.050302   \n",
       "fire_location_latitude         0.151646           0.030022    0.015118   \n",
       "fire_location_longitude        0.024582          -0.040349    0.040629   \n",
       "discovered_size                     NaN                NaN         NaN   \n",
       "assessment_hectares            0.019437          -0.010928    0.025223   \n",
       "fire_spread_rate               0.152724          -0.111232    0.134798   \n",
       "temperature                    1.000000          -0.274911   -0.017498   \n",
       "relative_humidity             -0.274911           1.000000   -0.168359   \n",
       "wind_speed                    -0.017498          -0.168359    1.000000   \n",
       "fire_fighting_start_size       0.025513          -0.023402    0.040925   \n",
       "distance_from_water_source    -0.027851           0.019332   -0.023366   \n",
       "bh_hectares                    0.017865          -0.020308    0.039520   \n",
       "uc_hectares                    0.017076          -0.018370    0.048538   \n",
       "to_hectares                    0.024049          -0.080456    0.122354   \n",
       "ex_hectares                    0.018138          -0.018491    0.050302   \n",
       "\n",
       "                            fire_fighting_start_size  \\\n",
       "fire_year                                   0.024184   \n",
       "current_size                                0.066583   \n",
       "fire_location_latitude                      0.049452   \n",
       "fire_location_longitude                     0.014862   \n",
       "discovered_size                                  NaN   \n",
       "assessment_hectares                         0.447678   \n",
       "fire_spread_rate                            0.120372   \n",
       "temperature                                 0.025513   \n",
       "relative_humidity                          -0.023402   \n",
       "wind_speed                                  0.040925   \n",
       "fire_fighting_start_size                    1.000000   \n",
       "distance_from_water_source                  0.001669   \n",
       "bh_hectares                                 0.029727   \n",
       "uc_hectares                                 0.061607   \n",
       "to_hectares                                 0.729295   \n",
       "ex_hectares                                 0.066583   \n",
       "\n",
       "                            distance_from_water_source  bh_hectares  \\\n",
       "fire_year                                    -0.006796     0.005634   \n",
       "current_size                                  0.002197     0.901745   \n",
       "fire_location_latitude                       -0.032771     0.014228   \n",
       "fire_location_longitude                      -0.003955     0.017869   \n",
       "discovered_size                                    NaN          NaN   \n",
       "assessment_hectares                          -0.001790     0.028499   \n",
       "fire_spread_rate                             -0.009101     0.069165   \n",
       "temperature                                  -0.027851     0.017865   \n",
       "relative_humidity                             0.019332    -0.020308   \n",
       "wind_speed                                   -0.023366     0.039520   \n",
       "fire_fighting_start_size                      0.001669     0.029727   \n",
       "distance_from_water_source                    1.000000     0.002453   \n",
       "bh_hectares                                   0.002453     1.000000   \n",
       "uc_hectares                                   0.002454     0.924775   \n",
       "to_hectares                                  -0.029221     0.663345   \n",
       "ex_hectares                                   0.002197     0.901745   \n",
       "\n",
       "                            uc_hectares  to_hectares  ex_hectares  \n",
       "fire_year                      0.008500     0.021763     0.008835  \n",
       "current_size                   0.997393     1.000000     1.000000  \n",
       "fire_location_latitude         0.022466     0.035165     0.024976  \n",
       "fire_location_longitude        0.014860    -0.033819     0.014599  \n",
       "discovered_size                     NaN          NaN          NaN  \n",
       "assessment_hectares            0.032869     0.578828     0.037687  \n",
       "fire_spread_rate               0.066523     0.176180     0.071773  \n",
       "temperature                    0.017076     0.024049     0.018138  \n",
       "relative_humidity             -0.018370    -0.080456    -0.018491  \n",
       "wind_speed                     0.048538     0.122354     0.050302  \n",
       "fire_fighting_start_size       0.061607     0.729295     0.066583  \n",
       "distance_from_water_source     0.002454    -0.029221     0.002197  \n",
       "bh_hectares                    0.924775     0.663345     0.901745  \n",
       "uc_hectares                    1.000000     0.802835     0.997393  \n",
       "to_hectares                    0.802835     1.000000     1.000000  \n",
       "ex_hectares                    0.997393     1.000000     1.000000  "
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "wildfires.select_dtypes(include='number').corr()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "66f0f265",
   "metadata": {},
   "outputs": [],
   "source": [
    "columns_that_describe_fire_size = ['current_size', 'bh_hectares', 'uc_hectares', 'ex_hectares']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "de156389",
   "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>current_size</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>ex_hectares</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.30</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20451</th>\n",
       "      <td>9.71</td>\n",
       "      <td>5.00</td>\n",
       "      <td>9.71</td>\n",
       "      <td>9.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20452</th>\n",
       "      <td>0.80</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1.00</td>\n",
       "      <td>0.80</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20453</th>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20454</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20455</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>20456 rows × 4 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       current_size  bh_hectares  uc_hectares  ex_hectares\n",
       "0              0.01         0.01         0.01         0.01\n",
       "1              0.01         0.01         0.01         0.01\n",
       "2              0.05         0.05         0.05         0.05\n",
       "3              0.15         0.15         0.15         0.15\n",
       "4              0.30         0.80         0.80         0.30\n",
       "...             ...          ...          ...          ...\n",
       "20451          9.71         5.00         9.71         9.71\n",
       "20452          0.80         1.00         1.00         0.80\n",
       "20453          0.10         0.10         0.10         0.10\n",
       "20454          0.05         0.05         0.05         0.05\n",
       "20455          0.01         0.01         0.01         0.01\n",
       "\n",
       "[20456 rows x 4 columns]"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "wildfires[columns_that_describe_fire_size]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "9aab570e",
   "metadata": {},
   "outputs": [],
   "source": [
    "test_df = wildfires[columns_that_describe_fire_size + ['size_class']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "166d73b4",
   "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>current_size</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>ex_hectares</th>\n",
       "      <th>size_class</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.30</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.30</td>\n",
       "      <td>B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20451</th>\n",
       "      <td>9.71</td>\n",
       "      <td>5.00</td>\n",
       "      <td>9.71</td>\n",
       "      <td>9.71</td>\n",
       "      <td>C</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20452</th>\n",
       "      <td>0.80</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1.00</td>\n",
       "      <td>0.80</td>\n",
       "      <td>B</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20453</th>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20454</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20455</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>A</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>20456 rows × 5 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       current_size  bh_hectares  uc_hectares  ex_hectares size_class\n",
       "0              0.01         0.01         0.01         0.01          A\n",
       "1              0.01         0.01         0.01         0.01          A\n",
       "2              0.05         0.05         0.05         0.05          A\n",
       "3              0.15         0.15         0.15         0.15          B\n",
       "4              0.30         0.80         0.80         0.30          B\n",
       "...             ...          ...          ...          ...        ...\n",
       "20451          9.71         5.00         9.71         9.71          C\n",
       "20452          0.80         1.00         1.00         0.80          B\n",
       "20453          0.10         0.10         0.10         0.10          A\n",
       "20454          0.05         0.05         0.05         0.05          A\n",
       "20455          0.01         0.01         0.01         0.01          A\n",
       "\n",
       "[20456 rows x 5 columns]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "45212b93",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.tree import DecisionTreeClassifier\n",
    "from sklearn.metrics import classification_report\n",
    "from sklearn import preprocessing"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "7545fc31",
   "metadata": {},
   "outputs": [],
   "source": [
    "label_encoder = preprocessing.LabelEncoder()\n",
    "size_labels = label_encoder.fit_transform(test_df['size_class'])\n",
    "test_df = test_df.drop('size_class', axis = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "833231f0",
   "metadata": {},
   "outputs": [],
   "source": [
    "X_train, X_test, y_train, y_test = train_test_split(test_df, size_labels, stratify=size_labels, test_size=0.2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "deee941a",
   "metadata": {},
   "outputs": [],
   "source": [
    "classifier = DecisionTreeClassifier(random_state = 42)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "d2b63d52",
   "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>current_size</th>\n",
       "      <th>bh_hectares</th>\n",
       "      <th>uc_hectares</th>\n",
       "      <th>ex_hectares</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.30</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20451</th>\n",
       "      <td>9.71</td>\n",
       "      <td>5.00</td>\n",
       "      <td>9.71</td>\n",
       "      <td>9.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20452</th>\n",
       "      <td>0.80</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1.00</td>\n",
       "      <td>0.80</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20453</th>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "      <td>0.10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20454</th>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.05</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20455</th>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.01</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>20456 rows × 4 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       current_size  bh_hectares  uc_hectares  ex_hectares\n",
       "0              0.01         0.01         0.01         0.01\n",
       "1              0.01         0.01         0.01         0.01\n",
       "2              0.05         0.05         0.05         0.05\n",
       "3              0.15         0.15         0.15         0.15\n",
       "4              0.30         0.80         0.80         0.30\n",
       "...             ...          ...          ...          ...\n",
       "20451          9.71         5.00         9.71         9.71\n",
       "20452          0.80         1.00         1.00         0.80\n",
       "20453          0.10         0.10         0.10         0.10\n",
       "20454          0.05         0.05         0.05         0.05\n",
       "20455          0.01         0.01         0.01         0.01\n",
       "\n",
       "[20456 rows x 4 columns]"
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "test_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "db727060",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<style>#sk-container-id-1 {color: black;background-color: white;}#sk-container-id-1 pre{padding: 0;}#sk-container-id-1 div.sk-toggleable {background-color: white;}#sk-container-id-1 label.sk-toggleable__label {cursor: pointer;display: block;width: 100%;margin-bottom: 0;padding: 0.3em;box-sizing: border-box;text-align: center;}#sk-container-id-1 label.sk-toggleable__label-arrow:before {content: \"▸\";float: left;margin-right: 0.25em;color: #696969;}#sk-container-id-1 label.sk-toggleable__label-arrow:hover:before {color: black;}#sk-container-id-1 div.sk-estimator:hover label.sk-toggleable__label-arrow:before {color: black;}#sk-container-id-1 div.sk-toggleable__content {max-height: 0;max-width: 0;overflow: hidden;text-align: left;background-color: #f0f8ff;}#sk-container-id-1 div.sk-toggleable__content pre {margin: 0.2em;color: black;border-radius: 0.25em;background-color: #f0f8ff;}#sk-container-id-1 input.sk-toggleable__control:checked~div.sk-toggleable__content {max-height: 200px;max-width: 100%;overflow: auto;}#sk-container-id-1 input.sk-toggleable__control:checked~label.sk-toggleable__label-arrow:before {content: \"▾\";}#sk-container-id-1 div.sk-estimator input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 div.sk-label input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 input.sk-hidden--visually {border: 0;clip: rect(1px 1px 1px 1px);clip: rect(1px, 1px, 1px, 1px);height: 1px;margin: -1px;overflow: hidden;padding: 0;position: absolute;width: 1px;}#sk-container-id-1 div.sk-estimator {font-family: monospace;background-color: #f0f8ff;border: 1px dotted black;border-radius: 0.25em;box-sizing: border-box;margin-bottom: 0.5em;}#sk-container-id-1 div.sk-estimator:hover {background-color: #d4ebff;}#sk-container-id-1 div.sk-parallel-item::after {content: \"\";width: 100%;border-bottom: 1px solid gray;flex-grow: 1;}#sk-container-id-1 div.sk-label:hover label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-1 div.sk-serial::before {content: \"\";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: 0;}#sk-container-id-1 div.sk-serial {display: flex;flex-direction: column;align-items: center;background-color: white;padding-right: 0.2em;padding-left: 0.2em;position: relative;}#sk-container-id-1 div.sk-item {position: relative;z-index: 1;}#sk-container-id-1 div.sk-parallel {display: flex;align-items: stretch;justify-content: center;background-color: white;position: relative;}#sk-container-id-1 div.sk-item::before, #sk-container-id-1 div.sk-parallel-item::before {content: \"\";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: -1;}#sk-container-id-1 div.sk-parallel-item {display: flex;flex-direction: column;z-index: 1;position: relative;background-color: white;}#sk-container-id-1 div.sk-parallel-item:first-child::after {align-self: flex-end;width: 50%;}#sk-container-id-1 div.sk-parallel-item:last-child::after {align-self: flex-start;width: 50%;}#sk-container-id-1 div.sk-parallel-item:only-child::after {width: 0;}#sk-container-id-1 div.sk-dashed-wrapped {border: 1px dashed gray;margin: 0 0.4em 0.5em 0.4em;box-sizing: border-box;padding-bottom: 0.4em;background-color: white;}#sk-container-id-1 div.sk-label label {font-family: monospace;font-weight: bold;display: inline-block;line-height: 1.2em;}#sk-container-id-1 div.sk-label-container {text-align: center;}#sk-container-id-1 div.sk-container {/* jupyter's `normalize.less` sets `[hidden] { display: none; }` but bootstrap.min.css set `[hidden] { display: none !important; }` so we also need the `!important` here to be able to override the default hidden behavior on the sphinx rendered scikit-learn.org. See: https://github.com/scikit-learn/scikit-learn/issues/21755 */display: inline-block !important;position: relative;}#sk-container-id-1 div.sk-text-repr-fallback {display: none;}</style><div id=\"sk-container-id-1\" class=\"sk-top-container\"><div class=\"sk-text-repr-fallback\"><pre>DecisionTreeClassifier(random_state=42)</pre><b>In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook. <br />On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.</b></div><div class=\"sk-container\" hidden><div class=\"sk-item\"><div class=\"sk-estimator sk-toggleable\"><input class=\"sk-toggleable__control sk-hidden--visually\" id=\"sk-estimator-id-1\" type=\"checkbox\" checked><label for=\"sk-estimator-id-1\" class=\"sk-toggleable__label sk-toggleable__label-arrow\">DecisionTreeClassifier</label><div class=\"sk-toggleable__content\"><pre>DecisionTreeClassifier(random_state=42)</pre></div></div></div></div></div>"
      ],
      "text/plain": [
       "DecisionTreeClassifier(random_state=42)"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "classifier.fit(X_train, y_train)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "id": "054678c1",
   "metadata": {},
   "outputs": [],
   "source": [
    "predictions = classifier.predict(X_test)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "id": "048f9931",
   "metadata": {},
   "outputs": [],
   "source": [
    "report = classification_report(y_test, predictions)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "id": "3cd43bc1",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       1.00      1.00      1.00      2590\n",
      "           1       1.00      1.00      1.00      1165\n",
      "           2       1.00      1.00      1.00       212\n",
      "           3       1.00      1.00      1.00        60\n",
      "           4       1.00      1.00      1.00        65\n",
      "\n",
      "    accuracy                           1.00      4092\n",
      "   macro avg       1.00      1.00      1.00      4092\n",
      "weighted avg       1.00      1.00      1.00      4092\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "id": "e3edac62",
   "metadata": {},
   "outputs": [],
   "source": [
    "feature_importance = classifier.feature_importances_"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "id": "eb5288cb",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([0.943911, 0.      , 0.      , 0.056089])"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "feature_importance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "id": "609fcc91",
   "metadata": {},
   "outputs": [],
   "source": [
    "# So current_size definetly needs to go"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "id": "ba28ae5d",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires = wildfires.drop('current_size', axis = 1)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "id": "0dedbcb0",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       1.00      1.00      1.00      2590\n",
      "           1       1.00      1.00      1.00      1165\n",
      "           2       1.00      1.00      1.00       212\n",
      "           3       1.00      0.98      0.99        60\n",
      "           4       0.98      1.00      0.99        65\n",
      "\n",
      "    accuracy                           1.00      4092\n",
      "   macro avg       1.00      1.00      1.00      4092\n",
      "weighted avg       1.00      1.00      1.00      4092\n",
      "\n"
     ]
    }
   ],
   "source": [
    "columns_that_describe_fire_size = ['bh_hectares', 'uc_hectares', 'ex_hectares']\n",
    "test_df = wildfires[columns_that_describe_fire_size + ['size_class']]\n",
    "label_encoder = preprocessing.LabelEncoder()\n",
    "size_labels = label_encoder.fit_transform(test_df['size_class'])\n",
    "test_df = test_df.drop('size_class', axis = 1)\n",
    "X_train, X_test, y_train, y_test = train_test_split(test_df, size_labels, stratify=size_labels, test_size=0.2)\n",
    "classifier = DecisionTreeClassifier(random_state = 42)\n",
    "classifier.fit(X_train, y_train)\n",
    "predictions = classifier.predict(X_test)\n",
    "report = classification_report(y_test, predictions)\n",
    "print(report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "id": "d45d9ea4",
   "metadata": {},
   "outputs": [],
   "source": [
    "feature_importance = classifier.feature_importances_"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "id": "359f5c5d",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([0., 0., 1.])"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "feature_importance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "id": "f77159cb",
   "metadata": {},
   "outputs": [],
   "source": [
    "# so ex_hectares also has to go "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "id": "92a2a576",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires = wildfires.drop('ex_hectares', axis = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "id": "a68769b9",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.98      0.99      0.99      2590\n",
      "           1       0.96      0.95      0.96      1165\n",
      "           2       0.87      0.83      0.85       212\n",
      "           3       0.83      0.75      0.79        60\n",
      "           4       0.95      0.92      0.94        65\n",
      "\n",
      "    accuracy                           0.97      4092\n",
      "   macro avg       0.92      0.89      0.90      4092\n",
      "weighted avg       0.97      0.97      0.97      4092\n",
      "\n"
     ]
    }
   ],
   "source": [
    "columns_that_describe_fire_size = ['bh_hectares', 'uc_hectares']\n",
    "test_df = wildfires[columns_that_describe_fire_size + ['size_class']]\n",
    "label_encoder = preprocessing.LabelEncoder()\n",
    "size_labels = label_encoder.fit_transform(test_df['size_class'])\n",
    "test_df = test_df.drop('size_class', axis = 1)\n",
    "X_train, X_test, y_train, y_test = train_test_split(test_df, size_labels, stratify=size_labels, test_size=0.2)\n",
    "classifier = DecisionTreeClassifier(random_state = 42)\n",
    "classifier.fit(X_train, y_train)\n",
    "predictions = classifier.predict(X_test)\n",
    "report = classification_report(y_test, predictions)\n",
    "print(report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "id": "1b798288",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([0.00939485, 0.99060515])"
      ]
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "feature_importance = classifier.feature_importances_\n",
    "feature_importance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "id": "db64e13a",
   "metadata": {},
   "outputs": [],
   "source": [
    "# And uc_hectares has to go as well \n",
    "wildfires = wildfires.drop('uc_hectares', axis = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "id": "aa4cbf0d",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.96      0.98      0.97      2590\n",
      "           1       0.90      0.91      0.91      1165\n",
      "           2       0.78      0.70      0.74       212\n",
      "           3       0.81      0.58      0.68        60\n",
      "           4       0.88      0.65      0.74        65\n",
      "\n",
      "    accuracy                           0.93      4092\n",
      "   macro avg       0.87      0.76      0.81      4092\n",
      "weighted avg       0.93      0.93      0.93      4092\n",
      "\n"
     ]
    }
   ],
   "source": [
    "columns_that_describe_fire_size = ['bh_hectares']\n",
    "test_df = wildfires[columns_that_describe_fire_size + ['size_class']]\n",
    "label_encoder = preprocessing.LabelEncoder()\n",
    "size_labels = label_encoder.fit_transform(test_df['size_class'])\n",
    "test_df = test_df.drop('size_class', axis = 1)\n",
    "X_train, X_test, y_train, y_test = train_test_split(test_df, size_labels, stratify=size_labels, test_size=0.2)\n",
    "classifier = DecisionTreeClassifier(random_state = 42)\n",
    "classifier.fit(X_train, y_train)\n",
    "predictions = classifier.predict(X_test)\n",
    "report = classification_report(y_test, predictions)\n",
    "print(report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "id": "71b10a81",
   "metadata": {},
   "outputs": [],
   "source": [
    "# let's remove bh_hectares as well to make the dataset simpler \n",
    "wildfires = wildfires.drop('bh_hectares', axis = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "id": "def4b6b0",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires = wildfires.drop('bh_fs_date', axis = 1)\n",
    "wildfires = wildfires.drop('uc_fs_date', axis = 1)\n",
    "wildfires = wildfires.drop('to_fs_date', axis = 1)\n",
    "wildfires = wildfires.drop('to_hectares', axis = 1)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "78e55d02",
   "metadata": {},
   "source": [
    "### I want to say that the dataset is ready but let's see "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "id": "7a35107a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 20456 entries, 0 to 20455\n",
      "Data columns (total 42 columns):\n",
      " #   Column                        Non-Null Count  Dtype         \n",
      "---  ------                        --------------  -----         \n",
      " 0   fire_year                     20456 non-null  int64         \n",
      " 1   fire_number                   20456 non-null  object        \n",
      " 2   fire_name                     591 non-null    object        \n",
      " 3   size_class                    20456 non-null  object        \n",
      " 4   fire_location_latitude        20456 non-null  float64       \n",
      " 5   fire_location_longitude       20456 non-null  float64       \n",
      " 6   fire_origin                   20447 non-null  object        \n",
      " 7   general_cause_desc            20456 non-null  object        \n",
      " 8   industry_identifier_desc      408 non-null    object        \n",
      " 9   responsible_group_desc        6577 non-null   object        \n",
      " 10  activity_class                11747 non-null  object        \n",
      " 11  true_cause                    10031 non-null  object        \n",
      " 12  fire_start_date               20300 non-null  object        \n",
      " 13  det_agent                     20455 non-null  object        \n",
      " 14  det_agent_type                20455 non-null  object        \n",
      " 15  discovered_date               16548 non-null  datetime64[ns]\n",
      " 16  discovered_size               0 non-null      float64       \n",
      " 17  reported_date                 20455 non-null  datetime64[ns]\n",
      " 18  dispatched_resource           20450 non-null  object        \n",
      " 19  dispatch_date                 20450 non-null  datetime64[ns]\n",
      " 20  start_for_fire_date           20449 non-null  datetime64[ns]\n",
      " 21  assessment_resource           20456 non-null  object        \n",
      " 22  assessment_datetime           20456 non-null  datetime64[ns]\n",
      " 23  assessment_hectares           20456 non-null  float64       \n",
      " 24  fire_spread_rate              20305 non-null  float64       \n",
      " 25  fire_type                     20437 non-null  object        \n",
      " 26  fire_position_on_slope        20265 non-null  object        \n",
      " 27  weather_conditions_over_fire  20258 non-null  object        \n",
      " 28  temperature                   20257 non-null  float64       \n",
      " 29  relative_humidity             20255 non-null  float64       \n",
      " 30  wind_direction                20254 non-null  object        \n",
      " 31  wind_speed                    20254 non-null  float64       \n",
      " 32  fuel_type                     16026 non-null  object        \n",
      " 33  initial_action_by             20449 non-null  object        \n",
      " 34  ia_arrival_at_fire_date       13979 non-null  datetime64[ns]\n",
      " 35  ia_access                     9576 non-null   object        \n",
      " 36  fire_fighting_start_date      14079 non-null  datetime64[ns]\n",
      " 37  fire_fighting_start_size      14079 non-null  float64       \n",
      " 38  bucketing_on_fire             13927 non-null  object        \n",
      " 39  distance_from_water_source    6365 non-null   float64       \n",
      " 40  first_bucket_drop_date        6365 non-null   datetime64[ns]\n",
      " 41  ex_fs_date                    20456 non-null  datetime64[ns]\n",
      "dtypes: datetime64[ns](9), float64(10), int64(1), object(22)\n",
      "memory usage: 6.6+ MB\n"
     ]
    }
   ],
   "source": [
    "wildfires.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "id": "ca0b8df1",
   "metadata": {},
   "outputs": [],
   "source": [
    "wildfires.to_csv('fp-historical-wildfire-data-2006-2021-modified.csv')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "24b1bd9c",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.10.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}