apply.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Subhalo Random Forest Classification"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Apply Random Forest Classifier to DMO Subhalo Catalogs\n",
    "\n",
    "This notebook applies a trained random forest to identify subhalos that are likely to be disrupted in hydrodynamic runs of dark matter only zoom-in simulations. After loading a DMO subhalo catalog with appropriate features and a trained random forest classifier, the classifier is used to output disruption probabilities."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Imports"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": false,
    "slideshow": {
     "slide_type": "fragment"
    }
   },
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pandas as pd\n",
    "import pickle"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Load Data\n",
    "\n",
    "The user should upload the DMO subhalo catalog they wish to classify. We provide an example corresponding to the code in $\\texttt{train.ipynb}$."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>$d_{peri}$</th>\n",
       "      <th>$a_{acc}$</th>\n",
       "      <th>$V_{acc}$</th>\n",
       "      <th>$M_{acc}$</th>\n",
       "      <th>$a_{peri}$</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>170.616243</td>\n",
       "      <td>0.8155</td>\n",
       "      <td>54.483148</td>\n",
       "      <td>1.713112e+10</td>\n",
       "      <td>0.9383</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>157.557844</td>\n",
       "      <td>0.9383</td>\n",
       "      <td>52.855967</td>\n",
       "      <td>1.372161e+10</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>192.665472</td>\n",
       "      <td>0.6910</td>\n",
       "      <td>55.918359</td>\n",
       "      <td>1.797514e+10</td>\n",
       "      <td>0.7849</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>89.381026</td>\n",
       "      <td>0.8916</td>\n",
       "      <td>41.694235</td>\n",
       "      <td>6.440465e+09</td>\n",
       "      <td>0.9748</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>50.607471</td>\n",
       "      <td>0.8473</td>\n",
       "      <td>34.527324</td>\n",
       "      <td>3.258255e+09</td>\n",
       "      <td>0.9383</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>177.577639</td>\n",
       "      <td>0.7750</td>\n",
       "      <td>28.082589</td>\n",
       "      <td>2.116738e+09</td>\n",
       "      <td>0.9030</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>154.628554</td>\n",
       "      <td>0.9383</td>\n",
       "      <td>25.385856</td>\n",
       "      <td>1.967154e+09</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>134.723775</td>\n",
       "      <td>0.8365</td>\n",
       "      <td>27.433545</td>\n",
       "      <td>1.877738e+09</td>\n",
       "      <td>0.9264</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>176.935008</td>\n",
       "      <td>0.8473</td>\n",
       "      <td>27.854052</td>\n",
       "      <td>1.546815e+09</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>38.838089</td>\n",
       "      <td>0.8473</td>\n",
       "      <td>32.086552</td>\n",
       "      <td>2.179413e+09</td>\n",
       "      <td>0.9146</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>37.894413</td>\n",
       "      <td>0.5929</td>\n",
       "      <td>45.167992</td>\n",
       "      <td>5.484465e+09</td>\n",
       "      <td>0.6566</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>182.853359</td>\n",
       "      <td>0.3700</td>\n",
       "      <td>53.632992</td>\n",
       "      <td>8.523776e+09</td>\n",
       "      <td>0.3994</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>46.068215</td>\n",
       "      <td>0.8803</td>\n",
       "      <td>26.300003</td>\n",
       "      <td>1.674671e+09</td>\n",
       "      <td>0.9503</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>163.183601</td>\n",
       "      <td>0.7179</td>\n",
       "      <td>27.186725</td>\n",
       "      <td>1.621189e+09</td>\n",
       "      <td>0.7849</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>130.765622</td>\n",
       "      <td>0.6083</td>\n",
       "      <td>26.382276</td>\n",
       "      <td>1.200014e+09</td>\n",
       "      <td>0.6822</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>76.502588</td>\n",
       "      <td>0.6320</td>\n",
       "      <td>28.951029</td>\n",
       "      <td>1.193329e+09</td>\n",
       "      <td>0.7088</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>20.148839</td>\n",
       "      <td>0.7554</td>\n",
       "      <td>31.803166</td>\n",
       "      <td>1.285252e+09</td>\n",
       "      <td>0.8365</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>206.900040</td>\n",
       "      <td>0.9383</td>\n",
       "      <td>26.510257</td>\n",
       "      <td>6.715399e+08</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>254.851077</td>\n",
       "      <td>0.8692</td>\n",
       "      <td>19.544458</td>\n",
       "      <td>6.598406e+08</td>\n",
       "      <td>0.9503</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>53.576682</td>\n",
       "      <td>0.8365</td>\n",
       "      <td>21.866391</td>\n",
       "      <td>8.440210e+08</td>\n",
       "      <td>0.9146</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>208.885215</td>\n",
       "      <td>0.9503</td>\n",
       "      <td>15.860447</td>\n",
       "      <td>6.289210e+08</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>62.162585</td>\n",
       "      <td>0.7554</td>\n",
       "      <td>24.782519</td>\n",
       "      <td>9.534930e+08</td>\n",
       "      <td>0.8473</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>63.681819</td>\n",
       "      <td>0.7849</td>\n",
       "      <td>23.319884</td>\n",
       "      <td>8.221266e+08</td>\n",
       "      <td>0.8692</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>256.112967</td>\n",
       "      <td>0.9873</td>\n",
       "      <td>20.467746</td>\n",
       "      <td>5.167748e+08</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>212.143638</td>\n",
       "      <td>0.9503</td>\n",
       "      <td>18.054399</td>\n",
       "      <td>5.429311e+08</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>83.478660</td>\n",
       "      <td>0.5563</td>\n",
       "      <td>26.629096</td>\n",
       "      <td>7.246045e+08</td>\n",
       "      <td>0.6320</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>25.367426</td>\n",
       "      <td>0.5635</td>\n",
       "      <td>35.596876</td>\n",
       "      <td>1.336227e+09</td>\n",
       "      <td>0.6240</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>56.283062</td>\n",
       "      <td>0.8582</td>\n",
       "      <td>20.933961</td>\n",
       "      <td>5.237944e+08</td>\n",
       "      <td>0.9503</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>40.562134</td>\n",
       "      <td>0.8582</td>\n",
       "      <td>23.676402</td>\n",
       "      <td>4.800056e+08</td>\n",
       "      <td>0.9748</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>243.774391</td>\n",
       "      <td>0.9748</td>\n",
       "      <td>17.222525</td>\n",
       "      <td>3.265776e+08</td>\n",
       "      <td>1.0000</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>1275</th>\n",
       "      <td>161.123638</td>\n",
       "      <td>0.9030</td>\n",
       "      <td>1.919708</td>\n",
       "      <td>2.120916e+06</td>\n",
       "      <td>0.9873</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1276</th>\n",
       "      <td>87.910374</td>\n",
       "      <td>0.5220</td>\n",
       "      <td>7.057213</td>\n",
       "      <td>2.120916e+07</td>\n",
       "      <td>0.5854</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1277</th>\n",
       "      <td>132.439581</td>\n",
       "      <td>0.7651</td>\n",
       "      <td>2.614460</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>0.8916</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1278</th>\n",
       "      <td>122.164123</td>\n",
       "      <td>0.9383</td>\n",
       "      <td>3.574314</td>\n",
       "      <td>2.592231e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1279</th>\n",
       "      <td>84.953012</td>\n",
       "      <td>0.7651</td>\n",
       "      <td>2.422489</td>\n",
       "      <td>2.120916e+06</td>\n",
       "      <td>0.8155</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1280</th>\n",
       "      <td>228.234202</td>\n",
       "      <td>0.9748</td>\n",
       "      <td>2.011123</td>\n",
       "      <td>2.238745e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1281</th>\n",
       "      <td>76.188968</td>\n",
       "      <td>0.8803</td>\n",
       "      <td>3.967397</td>\n",
       "      <td>3.534860e+06</td>\n",
       "      <td>0.9503</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1282</th>\n",
       "      <td>110.671757</td>\n",
       "      <td>0.5563</td>\n",
       "      <td>4.589017</td>\n",
       "      <td>7.304542e+06</td>\n",
       "      <td>0.6240</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1283</th>\n",
       "      <td>126.511503</td>\n",
       "      <td>0.7459</td>\n",
       "      <td>2.468196</td>\n",
       "      <td>1.885259e+06</td>\n",
       "      <td>0.7459</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1284</th>\n",
       "      <td>198.692846</td>\n",
       "      <td>0.7364</td>\n",
       "      <td>2.358499</td>\n",
       "      <td>3.299203e+06</td>\n",
       "      <td>0.8155</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1285</th>\n",
       "      <td>149.101181</td>\n",
       "      <td>0.8582</td>\n",
       "      <td>2.093396</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>0.8582</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1286</th>\n",
       "      <td>85.741554</td>\n",
       "      <td>0.8916</td>\n",
       "      <td>3.089816</td>\n",
       "      <td>2.592231e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1287</th>\n",
       "      <td>103.697095</td>\n",
       "      <td>0.9625</td>\n",
       "      <td>2.815572</td>\n",
       "      <td>3.181374e+06</td>\n",
       "      <td>0.9873</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1288</th>\n",
       "      <td>108.340607</td>\n",
       "      <td>0.9264</td>\n",
       "      <td>2.605318</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>0.9625</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1289</th>\n",
       "      <td>100.647733</td>\n",
       "      <td>0.7950</td>\n",
       "      <td>1.682030</td>\n",
       "      <td>9.426294e+05</td>\n",
       "      <td>0.8260</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1290</th>\n",
       "      <td>171.521750</td>\n",
       "      <td>0.9146</td>\n",
       "      <td>2.203094</td>\n",
       "      <td>1.649601e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1291</th>\n",
       "      <td>77.021049</td>\n",
       "      <td>0.9625</td>\n",
       "      <td>2.376782</td>\n",
       "      <td>1.885259e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1292</th>\n",
       "      <td>92.505061</td>\n",
       "      <td>0.8692</td>\n",
       "      <td>2.029406</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>0.9625</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1293</th>\n",
       "      <td>81.360305</td>\n",
       "      <td>0.6650</td>\n",
       "      <td>4.653007</td>\n",
       "      <td>5.420119e+06</td>\n",
       "      <td>0.7364</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1294</th>\n",
       "      <td>255.148545</td>\n",
       "      <td>0.9873</td>\n",
       "      <td>1.901425</td>\n",
       "      <td>2.120916e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1295</th>\n",
       "      <td>164.148684</td>\n",
       "      <td>0.8473</td>\n",
       "      <td>3.126382</td>\n",
       "      <td>2.592231e+06</td>\n",
       "      <td>0.9503</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1296</th>\n",
       "      <td>136.774945</td>\n",
       "      <td>0.9625</td>\n",
       "      <td>2.239660</td>\n",
       "      <td>2.238745e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1297</th>\n",
       "      <td>176.559497</td>\n",
       "      <td>0.9625</td>\n",
       "      <td>2.193952</td>\n",
       "      <td>2.120916e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1298</th>\n",
       "      <td>27.789034</td>\n",
       "      <td>0.7651</td>\n",
       "      <td>5.265485</td>\n",
       "      <td>1.037059e+07</td>\n",
       "      <td>0.8365</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1299</th>\n",
       "      <td>198.007855</td>\n",
       "      <td>0.9146</td>\n",
       "      <td>2.020264</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1300</th>\n",
       "      <td>259.505121</td>\n",
       "      <td>0.8473</td>\n",
       "      <td>1.700313</td>\n",
       "      <td>1.178287e+06</td>\n",
       "      <td>0.8582</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1301</th>\n",
       "      <td>65.445603</td>\n",
       "      <td>0.9146</td>\n",
       "      <td>1.928850</td>\n",
       "      <td>2.356573e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1302</th>\n",
       "      <td>44.977895</td>\n",
       "      <td>0.7554</td>\n",
       "      <td>3.108099</td>\n",
       "      <td>4.713147e+06</td>\n",
       "      <td>0.8260</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1303</th>\n",
       "      <td>41.046798</td>\n",
       "      <td>0.9030</td>\n",
       "      <td>1.709454</td>\n",
       "      <td>1.413944e+06</td>\n",
       "      <td>1.0000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1304</th>\n",
       "      <td>202.370219</td>\n",
       "      <td>0.7179</td>\n",
       "      <td>3.016684</td>\n",
       "      <td>4.477490e+06</td>\n",
       "      <td>0.7950</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1305 rows × 5 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      $d_{peri}$  $a_{acc}$  $V_{acc}$     $M_{acc}$  $a_{peri}$\n",
       "0     170.616243     0.8155  54.483148  1.713112e+10      0.9383\n",
       "1     157.557844     0.9383  52.855967  1.372161e+10      1.0000\n",
       "2     192.665472     0.6910  55.918359  1.797514e+10      0.7849\n",
       "3      89.381026     0.8916  41.694235  6.440465e+09      0.9748\n",
       "4      50.607471     0.8473  34.527324  3.258255e+09      0.9383\n",
       "5     177.577639     0.7750  28.082589  2.116738e+09      0.9030\n",
       "6     154.628554     0.9383  25.385856  1.967154e+09      1.0000\n",
       "7     134.723775     0.8365  27.433545  1.877738e+09      0.9264\n",
       "8     176.935008     0.8473  27.854052  1.546815e+09      1.0000\n",
       "9      38.838089     0.8473  32.086552  2.179413e+09      0.9146\n",
       "10     37.894413     0.5929  45.167992  5.484465e+09      0.6566\n",
       "11    182.853359     0.3700  53.632992  8.523776e+09      0.3994\n",
       "12     46.068215     0.8803  26.300003  1.674671e+09      0.9503\n",
       "13    163.183601     0.7179  27.186725  1.621189e+09      0.7849\n",
       "14    130.765622     0.6083  26.382276  1.200014e+09      0.6822\n",
       "15     76.502588     0.6320  28.951029  1.193329e+09      0.7088\n",
       "16     20.148839     0.7554  31.803166  1.285252e+09      0.8365\n",
       "17    206.900040     0.9383  26.510257  6.715399e+08      1.0000\n",
       "18    254.851077     0.8692  19.544458  6.598406e+08      0.9503\n",
       "19     53.576682     0.8365  21.866391  8.440210e+08      0.9146\n",
       "20    208.885215     0.9503  15.860447  6.289210e+08      1.0000\n",
       "21     62.162585     0.7554  24.782519  9.534930e+08      0.8473\n",
       "22     63.681819     0.7849  23.319884  8.221266e+08      0.8692\n",
       "23    256.112967     0.9873  20.467746  5.167748e+08      1.0000\n",
       "24    212.143638     0.9503  18.054399  5.429311e+08      1.0000\n",
       "25     83.478660     0.5563  26.629096  7.246045e+08      0.6320\n",
       "26     25.367426     0.5635  35.596876  1.336227e+09      0.6240\n",
       "27     56.283062     0.8582  20.933961  5.237944e+08      0.9503\n",
       "28     40.562134     0.8582  23.676402  4.800056e+08      0.9748\n",
       "29    243.774391     0.9748  17.222525  3.265776e+08      1.0000\n",
       "...          ...        ...        ...           ...         ...\n",
       "1275  161.123638     0.9030   1.919708  2.120916e+06      0.9873\n",
       "1276   87.910374     0.5220   7.057213  2.120916e+07      0.5854\n",
       "1277  132.439581     0.7651   2.614460  2.356573e+06      0.8916\n",
       "1278  122.164123     0.9383   3.574314  2.592231e+06      1.0000\n",
       "1279   84.953012     0.7651   2.422489  2.120916e+06      0.8155\n",
       "1280  228.234202     0.9748   2.011123  2.238745e+06      1.0000\n",
       "1281   76.188968     0.8803   3.967397  3.534860e+06      0.9503\n",
       "1282  110.671757     0.5563   4.589017  7.304542e+06      0.6240\n",
       "1283  126.511503     0.7459   2.468196  1.885259e+06      0.7459\n",
       "1284  198.692846     0.7364   2.358499  3.299203e+06      0.8155\n",
       "1285  149.101181     0.8582   2.093396  2.356573e+06      0.8582\n",
       "1286   85.741554     0.8916   3.089816  2.592231e+06      1.0000\n",
       "1287  103.697095     0.9625   2.815572  3.181374e+06      0.9873\n",
       "1288  108.340607     0.9264   2.605318  2.356573e+06      0.9625\n",
       "1289  100.647733     0.7950   1.682030  9.426294e+05      0.8260\n",
       "1290  171.521750     0.9146   2.203094  1.649601e+06      1.0000\n",
       "1291   77.021049     0.9625   2.376782  1.885259e+06      1.0000\n",
       "1292   92.505061     0.8692   2.029406  2.356573e+06      0.9625\n",
       "1293   81.360305     0.6650   4.653007  5.420119e+06      0.7364\n",
       "1294  255.148545     0.9873   1.901425  2.120916e+06      1.0000\n",
       "1295  164.148684     0.8473   3.126382  2.592231e+06      0.9503\n",
       "1296  136.774945     0.9625   2.239660  2.238745e+06      1.0000\n",
       "1297  176.559497     0.9625   2.193952  2.120916e+06      1.0000\n",
       "1298   27.789034     0.7651   5.265485  1.037059e+07      0.8365\n",
       "1299  198.007855     0.9146   2.020264  2.356573e+06      1.0000\n",
       "1300  259.505121     0.8473   1.700313  1.178287e+06      0.8582\n",
       "1301   65.445603     0.9146   1.928850  2.356573e+06      1.0000\n",
       "1302   44.977895     0.7554   3.108099  4.713147e+06      0.8260\n",
       "1303   41.046798     0.9030   1.709454  1.413944e+06      1.0000\n",
       "1304  202.370219     0.7179   3.016684  4.477490e+06      0.7950\n",
       "\n",
       "[1305 rows x 5 columns]"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "Halo023_properties = np.loadtxt('Halo023_properties_example.txt')\n",
    "\n",
    "Halo023_features = pd.DataFrame(Halo023_properties)\n",
    "Halo023_features = Halo023_features.rename(index=str, columns={0:\"$d_{peri}$\", \\\n",
    "                                                       1: \"$a_{acc}$\", \\\n",
    "                                                       2: \"$V_{acc}$\", \\\n",
    "                                                       3: \"$M_{acc}$\", \\\n",
    "                                                       4: \"$a_{peri}$\"})\n",
    "\n",
    "Halo023_features"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Load finalized model\n",
    "filename = 'finalized_rf.sav'\n",
    "loaded_model = pickle.load(open(filename, 'rb'))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "slideshow": {
     "slide_type": "slide"
    }
   },
   "source": [
    "### Apply Random Forest Classifier"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {
    "collapsed": false,
    "slideshow": {
     "slide_type": "-"
    }
   },
   "outputs": [],
   "source": [
    "#Survival and Disruption Probabilities\n",
    "prob = loaded_model.predict_proba(Halo023_features)\n",
    "surv_prob = prob[:,0]\n",
    "dest_prob = prob[:,1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Most likely classification (50% probability split)\n",
    "pred = loaded_model.predict(Halo023_features)\n",
    "Halo023_surviving_features = Halo023_features[pred==0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Survival Fraction: 0.78\n"
     ]
    }
   ],
   "source": [
    "print('Survival Fraction: %0.2f' % (len(pred[pred==0])/len(pred)))"
   ]
  }
 ],
 "metadata": {
  "anaconda-cloud": {},
  "celltoolbar": "Slideshow",
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.5.2"
  },
  "livereveal": {
   "scroll": true,
   "start_slideshow_at": "selected"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}