Predict_Cabin_Class v0.3.ipynb

{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "##### Predicting reviewer‘s cabin class based on their written review on Skytrax"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import scipy as sp\n",
    "import matplotlib.pyplot as plt\n",
    "import pickle\n",
    "from sklearn.metrics import classification_report, confusion_matrix\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer\n",
    "from textacy.preprocess import preprocess_text\n",
    "from sklearn.naive_bayes import MultinomialNB\n",
    "from sklearn.linear_model import LogisticRegression\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.neighbors import KNeighborsClassifier\n",
    "from sklearn import metrics\n",
    "from textblob import TextBlob, Word\n",
    "from nltk.stem.snowball import SnowballStemmer\n",
    "\n",
    "%config InlineBackend.figure_format = 'retina'\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "# we open the file for reading\n",
    "fileObject1 = open(\"df_air.pickle\",'rb')\n",
    "fileObject2 = open(\"df_main.pickle\",'rb')\n",
    "fileObject3 = open(\"star.pickle\",'rb')\n",
    "fileObject4 = open(\"budget.pickle\",'rb')\n",
    "\n",
    "# load the object from the file into variables\n",
    "df_air = pickle.load(fileObject1)  \n",
    "df = pickle.load(fileObject2) \n",
    "star = pickle.load(fileObject3) \n",
    "budget = pickle.load(fileObject4) "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "scrolled": true
   },
   "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>airline_name</th>\n",
       "      <th>author_country</th>\n",
       "      <th>content</th>\n",
       "      <th>cabin_flown</th>\n",
       "      <th>overall_rating</th>\n",
       "      <th>seat_comfort_rating</th>\n",
       "      <th>cabin_staff_rating</th>\n",
       "      <th>food_beverages_rating</th>\n",
       "      <th>inflight_entertainment_rating</th>\n",
       "      <th>ground_service_rating</th>\n",
       "      <th>wifi_connectivity_rating</th>\n",
       "      <th>value_money_rating</th>\n",
       "      <th>recommend</th>\n",
       "      <th>airline</th>\n",
       "      <th>cabin_class</th>\n",
       "      <th>overall</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>54</td>\n",
       "      <td>Outbound flight FRA/PRN A319. 2 hours 10 min f...</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>149</td>\n",
       "      <td>Two short hops ZRH-LJU and LJU-VIE. Very fast ...</td>\n",
       "      <td>0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>138</td>\n",
       "      <td>Flew Zurich-Ljubljana on JP365 newish CRJ900. ...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0</td>\n",
       "      <td>129</td>\n",
       "      <td>Adria serves this 100 min flight from Ljubljan...</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>118</td>\n",
       "      <td>WAW-SKJ Economy. No free snacks or drinks on t...</td>\n",
       "      <td>1</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>0</td>\n",
       "      <td>54</td>\n",
       "      <td>Sarajevo-Frankfurt via Ljubljana. I loved flyi...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>0</td>\n",
       "      <td>51</td>\n",
       "      <td>I had flights from Paris to Sarajevo via Ljubl...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>0</td>\n",
       "      <td>131</td>\n",
       "      <td>LJU to FRA and back both flights were on time....</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>0</td>\n",
       "      <td>129</td>\n",
       "      <td>On my Ljubljana - Munich flight in business cl...</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>0</td>\n",
       "      <td>131</td>\n",
       "      <td>Flights from LJU to ZRH and back all on time. ...</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>0</td>\n",
       "      <td>144</td>\n",
       "      <td>I was very satisfied with the CRJ 900 on my fl...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>0</td>\n",
       "      <td>89</td>\n",
       "      <td>I was on JP650 the evening departure to Istanb...</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>0</td>\n",
       "      <td>27</td>\n",
       "      <td>VIE-LJU LJU-MUC Aug 4-15th CRJ200. Both flight...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0</td>\n",
       "      <td>129</td>\n",
       "      <td>If I have to fly a regional jet then I prefer ...</td>\n",
       "      <td>0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>0</td>\n",
       "      <td>144</td>\n",
       "      <td>Istanbul-Ljubljana-Munich and return. Pre-take...</td>\n",
       "      <td>0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>0</td>\n",
       "      <td>51</td>\n",
       "      <td>Return flight Paris-Skopje via Ljubljana. All ...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>0</td>\n",
       "      <td>57</td>\n",
       "      <td>BEG-LJU-BEG with CRJ200 great regional airline...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>0</td>\n",
       "      <td>138</td>\n",
       "      <td>I fly at least once a month ZRH-LJU-ZRH. My ex...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>0</td>\n",
       "      <td>148</td>\n",
       "      <td>LTN to LJU outbound with CRJ200 return with CR...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>0</td>\n",
       "      <td>131</td>\n",
       "      <td>Ljubljana to Brussels with CRJ900. Boarding on...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>0</td>\n",
       "      <td>131</td>\n",
       "      <td>LJU-FRA and FRA-LJU. First flight on time retu...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>0</td>\n",
       "      <td>131</td>\n",
       "      <td>LJU-FRA-LJU. Both flights on time. Flight atte...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>0</td>\n",
       "      <td>7</td>\n",
       "      <td>Smart and efficient service. Small planes but ...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>0</td>\n",
       "      <td>27</td>\n",
       "      <td>FRA-VIE in Business Class. A great little airl...</td>\n",
       "      <td>0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>0</td>\n",
       "      <td>148</td>\n",
       "      <td>London Gatwick to Ljubljana and Ljubljana to V...</td>\n",
       "      <td>0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>0</td>\n",
       "      <td>148</td>\n",
       "      <td>ZRH to LJU as a continuation to a flight with ...</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>0</td>\n",
       "      <td>148</td>\n",
       "      <td>I fly Adria a lot (at least once a month) and ...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>1</td>\n",
       "      <td>149</td>\n",
       "      <td>Brand new A320, interior spotless. Crew immacu...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>0</td>\n",
       "      <td>135</td>\n",
       "      <td>I fly Adria quite a lot (aprox 2 flts a month)...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>0</td>\n",
       "      <td>146</td>\n",
       "      <td>This was my first time with Adria. Short hop f...</td>\n",
       "      <td>1</td>\n",
       "      <td>6.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>3.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>0</td>\n",
       "      <td>138</td>\n",
       "      <td>ZRH-LJU-ZRH with CRJ-900. New plane with pleas...</td>\n",
       "      <td>1</td>\n",
       "      <td>7.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>3.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>0</td>\n",
       "      <td>87</td>\n",
       "      <td>It is reliable and with good connections betwe...</td>\n",
       "      <td>1</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>adria-airways</td>\n",
       "      <td>Economy</td>\n",
       "      <td>2.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>1</td>\n",
       "      <td>7</td>\n",
       "      <td>Flight to Larnaca was a joy. Generous legroom,...</td>\n",
       "      <td>0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>1</td>\n",
       "      <td>148</td>\n",
       "      <td>Flight on time, row 1 - Friendly staff and ver...</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>1</td>\n",
       "      <td>15</td>\n",
       "      <td>Very good flight with Aegean. The boarding was...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>1</td>\n",
       "      <td>57</td>\n",
       "      <td>I booked a premium economy ticket so I can hav...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>1</td>\n",
       "      <td>149</td>\n",
       "      <td>2 flights - HER to ATH (short 50 minutes) and ...</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>1</td>\n",
       "      <td>72</td>\n",
       "      <td>Managed to get my rather large luggage quickly...</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>1</td>\n",
       "      <td>138</td>\n",
       "      <td>I recently flew to Mykonos through Athens and ...</td>\n",
       "      <td>0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>1</td>\n",
       "      <td>148</td>\n",
       "      <td>We flew on flights 4441 and 4440 on 22nd June ...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>1</td>\n",
       "      <td>54</td>\n",
       "      <td>I am always impressed with the pleasant attitu...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>1</td>\n",
       "      <td>72</td>\n",
       "      <td>A good flight, on time. Flight attendants were...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>1</td>\n",
       "      <td>27</td>\n",
       "      <td>We flew Heathrow-Athens return and Athens-Sant...</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>1</td>\n",
       "      <td>137</td>\n",
       "      <td>Very good flight, nice to be offered a hot mea...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>1</td>\n",
       "      <td>148</td>\n",
       "      <td>Return LHR-ATH.  Clean, plane seemed quite new...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>1</td>\n",
       "      <td>27</td>\n",
       "      <td>Multiple trip with AE between SKG-FRA and MUC....</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>1</td>\n",
       "      <td>148</td>\n",
       "      <td>Flew from Athens to Santorini and back. Flight...</td>\n",
       "      <td>1</td>\n",
       "      <td>8.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>1</td>\n",
       "      <td>57</td>\n",
       "      <td>Athens to Corfu on Olympic Air Bombardier Dash...</td>\n",
       "      <td>1</td>\n",
       "      <td>10.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>1</td>\n",
       "      <td>148</td>\n",
       "      <td>LHR-ATH return. Plenty of legroom - better tha...</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Economy</td>\n",
       "      <td>4.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>1</td>\n",
       "      <td>138</td>\n",
       "      <td>Traveled on 27 May from Zurich to Larnaca on b...</td>\n",
       "      <td>0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1</td>\n",
       "      <td>aegean-airlines</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    airline_name  author_country  \\\n",
       "0              0              54   \n",
       "1              0             149   \n",
       "2              0             138   \n",
       "3              0             129   \n",
       "4              0             118   \n",
       "5              0              54   \n",
       "6              0              51   \n",
       "7              0             131   \n",
       "8              0             129   \n",
       "9              0             131   \n",
       "10             0             144   \n",
       "11             0              89   \n",
       "12             0              27   \n",
       "13             0             129   \n",
       "14             0             144   \n",
       "15             0              51   \n",
       "16             0              57   \n",
       "17             0             138   \n",
       "18             0             148   \n",
       "19             0             131   \n",
       "20             0             131   \n",
       "21             0             131   \n",
       "22             0               7   \n",
       "23             0              27   \n",
       "24             0             148   \n",
       "25             0             148   \n",
       "26             0             148   \n",
       "27             1             149   \n",
       "28             0             135   \n",
       "29             0             146   \n",
       "30             0             138   \n",
       "31             0              87   \n",
       "32             1               7   \n",
       "33             1             148   \n",
       "34             1              15   \n",
       "35             1              57   \n",
       "36             1             149   \n",
       "37             1              72   \n",
       "38             1             138   \n",
       "39             1             148   \n",
       "40             1              54   \n",
       "41             1              72   \n",
       "42             1              27   \n",
       "43             1             137   \n",
       "44             1             148   \n",
       "45             1              27   \n",
       "46             1             148   \n",
       "47             1              57   \n",
       "48             1             148   \n",
       "49             1             138   \n",
       "\n",
       "                                              content  cabin_flown  \\\n",
       "0   Outbound flight FRA/PRN A319. 2 hours 10 min f...            1   \n",
       "1   Two short hops ZRH-LJU and LJU-VIE. Very fast ...            0   \n",
       "2   Flew Zurich-Ljubljana on JP365 newish CRJ900. ...            1   \n",
       "3   Adria serves this 100 min flight from Ljubljan...            0   \n",
       "4   WAW-SKJ Economy. No free snacks or drinks on t...            1   \n",
       "5   Sarajevo-Frankfurt via Ljubljana. I loved flyi...            1   \n",
       "6   I had flights from Paris to Sarajevo via Ljubl...            1   \n",
       "7   LJU to FRA and back both flights were on time....            1   \n",
       "8   On my Ljubljana - Munich flight in business cl...            0   \n",
       "9   Flights from LJU to ZRH and back all on time. ...            1   \n",
       "10  I was very satisfied with the CRJ 900 on my fl...            1   \n",
       "11  I was on JP650 the evening departure to Istanb...            1   \n",
       "12  VIE-LJU LJU-MUC Aug 4-15th CRJ200. Both flight...            1   \n",
       "13  If I have to fly a regional jet then I prefer ...            0   \n",
       "14  Istanbul-Ljubljana-Munich and return. Pre-take...            0   \n",
       "15  Return flight Paris-Skopje via Ljubljana. All ...            1   \n",
       "16  BEG-LJU-BEG with CRJ200 great regional airline...            1   \n",
       "17  I fly at least once a month ZRH-LJU-ZRH. My ex...            1   \n",
       "18  LTN to LJU outbound with CRJ200 return with CR...            1   \n",
       "19  Ljubljana to Brussels with CRJ900. Boarding on...            1   \n",
       "20  LJU-FRA and FRA-LJU. First flight on time retu...            1   \n",
       "21  LJU-FRA-LJU. Both flights on time. Flight atte...            1   \n",
       "22  Smart and efficient service. Small planes but ...            1   \n",
       "23  FRA-VIE in Business Class. A great little airl...            0   \n",
       "24  London Gatwick to Ljubljana and Ljubljana to V...            0   \n",
       "25  ZRH to LJU as a continuation to a flight with ...            1   \n",
       "26  I fly Adria a lot (at least once a month) and ...            1   \n",
       "27  Brand new A320, interior spotless. Crew immacu...            1   \n",
       "28  I fly Adria quite a lot (aprox 2 flts a month)...            1   \n",
       "29  This was my first time with Adria. Short hop f...            1   \n",
       "30  ZRH-LJU-ZRH with CRJ-900. New plane with pleas...            1   \n",
       "31  It is reliable and with good connections betwe...            1   \n",
       "32  Flight to Larnaca was a joy. Generous legroom,...            0   \n",
       "33  Flight on time, row 1 - Friendly staff and ver...            0   \n",
       "34  Very good flight with Aegean. The boarding was...            1   \n",
       "35  I booked a premium economy ticket so I can hav...            1   \n",
       "36  2 flights - HER to ATH (short 50 minutes) and ...            1   \n",
       "37  Managed to get my rather large luggage quickly...            1   \n",
       "38  I recently flew to Mykonos through Athens and ...            0   \n",
       "39  We flew on flights 4441 and 4440 on 22nd June ...            1   \n",
       "40  I am always impressed with the pleasant attitu...            1   \n",
       "41  A good flight, on time. Flight attendants were...            1   \n",
       "42  We flew Heathrow-Athens return and Athens-Sant...            1   \n",
       "43  Very good flight, nice to be offered a hot mea...            1   \n",
       "44  Return LHR-ATH.  Clean, plane seemed quite new...            1   \n",
       "45  Multiple trip with AE between SKG-FRA and MUC....            1   \n",
       "46  Flew from Athens to Santorini and back. Flight...            1   \n",
       "47  Athens to Corfu on Olympic Air Bombardier Dash...            1   \n",
       "48  LHR-ATH return. Plenty of legroom - better tha...            1   \n",
       "49  Traveled on 27 May from Zurich to Larnaca on b...            0   \n",
       "\n",
       "    overall_rating  seat_comfort_rating  cabin_staff_rating  \\\n",
       "0              7.0                  4.0                 4.0   \n",
       "1             10.0                  4.0                 5.0   \n",
       "2              9.0                  5.0                 5.0   \n",
       "3              8.0                  4.0                 4.0   \n",
       "4              4.0                  4.0                 2.0   \n",
       "5              9.0                  4.0                 4.0   \n",
       "6              5.0                  4.0                 4.0   \n",
       "7              9.0                  5.0                 5.0   \n",
       "8              8.0                  4.0                 3.0   \n",
       "9             10.0                  5.0                 5.0   \n",
       "10             9.0                  4.0                 4.0   \n",
       "11             7.0                  4.0                 5.0   \n",
       "12             8.0                  4.0                 4.0   \n",
       "13             7.0                  3.0                 3.0   \n",
       "14             7.0                  2.0                 5.0   \n",
       "15             5.0                  4.0                 4.0   \n",
       "16             9.0                  4.0                 5.0   \n",
       "17             5.0                  3.0                 3.0   \n",
       "18             5.0                  3.0                 3.0   \n",
       "19             8.0                  3.0                 3.0   \n",
       "20             9.0                  3.0                 3.0   \n",
       "21             5.0                  3.0                 3.0   \n",
       "22             9.0                  3.0                 3.0   \n",
       "23             9.0                  3.0                 3.0   \n",
       "24            10.0                  3.0                 3.0   \n",
       "25             7.0                  3.0                 3.0   \n",
       "26             5.0                  3.0                 3.0   \n",
       "27             8.0                  4.0                 4.0   \n",
       "28             5.0                  3.0                 3.0   \n",
       "29             6.0                  3.0                 3.0   \n",
       "30             7.0                  3.0                 3.0   \n",
       "31             5.0                  3.0                 3.0   \n",
       "32            10.0                  5.0                 5.0   \n",
       "33             8.0                  4.0                 5.0   \n",
       "34             9.0                  4.0                 5.0   \n",
       "35             8.0                  4.0                 5.0   \n",
       "36            10.0                  4.0                 5.0   \n",
       "37            10.0                  4.0                 5.0   \n",
       "38            10.0                  5.0                 5.0   \n",
       "39             9.0                  4.0                 5.0   \n",
       "40             8.0                  4.0                 5.0   \n",
       "41             8.0                  4.0                 5.0   \n",
       "42            10.0                  5.0                 5.0   \n",
       "43             9.0                  4.0                 4.0   \n",
       "44             9.0                  4.0                 4.0   \n",
       "45            10.0                  5.0                 5.0   \n",
       "46             8.0                  4.0                 5.0   \n",
       "47            10.0                  4.0                 5.0   \n",
       "48             9.0                  4.0                 4.0   \n",
       "49            10.0                  4.0                 5.0   \n",
       "\n",
       "    food_beverages_rating  inflight_entertainment_rating  \\\n",
       "0                     4.0                            0.0   \n",
       "1                     4.0                            1.0   \n",
       "2                     4.0                            0.0   \n",
       "3                     3.0                            1.0   \n",
       "4                     1.0                            2.0   \n",
       "5                     3.0                            3.0   \n",
       "6                     1.0                            0.0   \n",
       "7                     4.0                            3.0   \n",
       "8                     4.0                            1.0   \n",
       "9                     4.0                            4.0   \n",
       "10                    4.0                            4.0   \n",
       "11                    3.0                            3.0   \n",
       "12                    4.0                            0.0   \n",
       "13                    4.0                            1.0   \n",
       "14                    4.0                            0.0   \n",
       "15                    1.0                            0.0   \n",
       "16                    4.0                            0.0   \n",
       "17                    3.0                            3.0   \n",
       "18                    3.0                            3.0   \n",
       "19                    3.0                            3.0   \n",
       "20                    3.0                            3.0   \n",
       "21                    3.0                            3.0   \n",
       "22                    3.0                            3.0   \n",
       "23                    3.0                            3.0   \n",
       "24                    3.0                            3.0   \n",
       "25                    3.0                            3.0   \n",
       "26                    3.0                            3.0   \n",
       "27                    3.0                            3.0   \n",
       "28                    3.0                            3.0   \n",
       "29                    3.0                            3.0   \n",
       "30                    3.0                            3.0   \n",
       "31                    3.0                            3.0   \n",
       "32                    5.0                            3.0   \n",
       "33                    4.0                            2.0   \n",
       "34                    4.0                            3.0   \n",
       "35                    4.0                            3.0   \n",
       "36                    4.0                            3.0   \n",
       "37                    5.0                            3.0   \n",
       "38                    5.0                            3.0   \n",
       "39                    3.0                            2.0   \n",
       "40                    4.0                            3.0   \n",
       "41                    5.0                            3.0   \n",
       "42                    4.0                            3.0   \n",
       "43                    4.0                            3.0   \n",
       "44                    4.0                            3.0   \n",
       "45                    4.0                            3.0   \n",
       "46                    4.0                            0.0   \n",
       "47                    5.0                            3.0   \n",
       "48                    3.0                            0.0   \n",
       "49                    5.0                            0.0   \n",
       "\n",
       "    ground_service_rating  wifi_connectivity_rating  value_money_rating  \\\n",
       "0                     3.0                       3.0                 4.0   \n",
       "1                     3.0                       3.0                 5.0   \n",
       "2                     3.0                       3.0                 5.0   \n",
       "3                     3.0                       3.0                 4.0   \n",
       "4                     3.0                       3.0                 2.0   \n",
       "5                     3.0                       3.0                 4.0   \n",
       "6                     3.0                       3.0                 3.0   \n",
       "7                     3.0                       3.0                 4.0   \n",
       "8                     3.0                       3.0                 4.0   \n",
       "9                     3.0                       3.0                 4.0   \n",
       "10                    3.0                       3.0                 4.0   \n",
       "11                    3.0                       3.0                 5.0   \n",
       "12                    3.0                       3.0                 4.0   \n",
       "13                    3.0                       3.0                 4.0   \n",
       "14                    3.0                       3.0                 4.0   \n",
       "15                    3.0                       3.0                 4.0   \n",
       "16                    3.0                       3.0                 2.0   \n",
       "17                    3.0                       3.0                 4.0   \n",
       "18                    3.0                       3.0                 4.0   \n",
       "19                    3.0                       3.0                 1.0   \n",
       "20                    3.0                       3.0                 4.0   \n",
       "21                    3.0                       3.0                 1.0   \n",
       "22                    3.0                       3.0                 4.0   \n",
       "23                    3.0                       3.0                 4.0   \n",
       "24                    3.0                       3.0                 4.0   \n",
       "25                    3.0                       3.0                 1.0   \n",
       "26                    3.0                       3.0                 1.0   \n",
       "27                    4.0                       3.0                 4.0   \n",
       "28                    3.0                       3.0                 1.0   \n",
       "29                    3.0                       3.0                 1.0   \n",
       "30                    3.0                       3.0                 4.0   \n",
       "31                    3.0                       3.0                 4.0   \n",
       "32                    3.0                       3.0                 5.0   \n",
       "33                    4.0                       3.0                 4.0   \n",
       "34                    4.0                       3.0                 4.0   \n",
       "35                    2.0                       3.0                 3.0   \n",
       "36                    5.0                       3.0                 5.0   \n",
       "37                    5.0                       3.0                 5.0   \n",
       "38                    4.0                       3.0                 5.0   \n",
       "39                    5.0                       2.0                 5.0   \n",
       "40                    4.0                       3.0                 5.0   \n",
       "41                    5.0                       3.0                 5.0   \n",
       "42                    4.0                       3.0                 5.0   \n",
       "43                    4.0                       3.0                 5.0   \n",
       "44                    3.0                       3.0                 4.0   \n",
       "45                    3.0                       3.0                 4.0   \n",
       "46                    3.0                       3.0                 4.0   \n",
       "47                    3.0                       3.0                 4.0   \n",
       "48                    3.0                       3.0                 4.0   \n",
       "49                    3.0                       3.0                 5.0   \n",
       "\n",
       "    recommend          airline     cabin_class  overall  \n",
       "0           1    adria-airways         Economy      3.5  \n",
       "1           1    adria-airways  Business Class      5.0  \n",
       "2           1    adria-airways         Economy      4.5  \n",
       "3           1    adria-airways  Business Class      4.0  \n",
       "4           0    adria-airways         Economy      2.0  \n",
       "5           1    adria-airways         Economy      4.5  \n",
       "6           1    adria-airways         Economy      2.5  \n",
       "7           1    adria-airways         Economy      4.5  \n",
       "8           1    adria-airways  Business Class      4.0  \n",
       "9           1    adria-airways         Economy      5.0  \n",
       "10          1    adria-airways         Economy      4.5  \n",
       "11          1    adria-airways         Economy      3.5  \n",
       "12          1    adria-airways         Economy      4.0  \n",
       "13          1    adria-airways  Business Class      3.5  \n",
       "14          1    adria-airways  Business Class      3.5  \n",
       "15          1    adria-airways         Economy      2.5  \n",
       "16          1    adria-airways         Economy      4.5  \n",
       "17          1    adria-airways         Economy      2.5  \n",
       "18          1    adria-airways         Economy      2.5  \n",
       "19          1    adria-airways         Economy      4.0  \n",
       "20          1    adria-airways         Economy      4.5  \n",
       "21          0    adria-airways         Economy      2.5  \n",
       "22          1    adria-airways         Economy      4.5  \n",
       "23          1    adria-airways  Business Class      4.5  \n",
       "24          1    adria-airways  Business Class      5.0  \n",
       "25          1    adria-airways         Economy      3.5  \n",
       "26          0    adria-airways         Economy      2.5  \n",
       "27          1  aegean-airlines         Economy      4.0  \n",
       "28          0    adria-airways         Economy      2.5  \n",
       "29          1    adria-airways         Economy      3.0  \n",
       "30          1    adria-airways         Economy      3.5  \n",
       "31          1    adria-airways         Economy      2.5  \n",
       "32          1  aegean-airlines  Business Class      5.0  \n",
       "33          1  aegean-airlines  Business Class      4.0  \n",
       "34          1  aegean-airlines         Economy      4.5  \n",
       "35          1  aegean-airlines         Economy      4.0  \n",
       "36          1  aegean-airlines         Economy      5.0  \n",
       "37          1  aegean-airlines         Economy      5.0  \n",
       "38          1  aegean-airlines  Business Class      5.0  \n",
       "39          1  aegean-airlines         Economy      4.5  \n",
       "40          1  aegean-airlines         Economy      4.0  \n",
       "41          1  aegean-airlines         Economy      4.0  \n",
       "42          1  aegean-airlines         Economy      5.0  \n",
       "43          1  aegean-airlines         Economy      4.5  \n",
       "44          1  aegean-airlines         Economy      4.5  \n",
       "45          1  aegean-airlines         Economy      5.0  \n",
       "46          1  aegean-airlines         Economy      4.0  \n",
       "47          1  aegean-airlines         Economy      5.0  \n",
       "48          1  aegean-airlines         Economy      4.5  \n",
       "49          1  aegean-airlines  Business Class      5.0  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.reset_index(inplace=True)\n",
    "df.drop(columns=['index'],axis=1,inplace=True)\n",
    "df.head(50)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['airline_name', 'author_country', 'content', 'cabin_flown',\n",
       "       'overall_rating', 'seat_comfort_rating', 'cabin_staff_rating',\n",
       "       'food_beverages_rating', 'inflight_entertainment_rating',\n",
       "       'ground_service_rating', 'wifi_connectivity_rating',\n",
       "       'value_money_rating', 'recommend', 'airline', 'cabin_class', 'overall'],\n",
       "      dtype='object')"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "y = df['cabin_flown']\n",
    "#.astype('uint8')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "clean_text = [preprocess_text(x, fix_unicode=True, lowercase=True, transliterate=False,\n",
    "                              no_urls=True, no_emails=True, no_phone_numbers=True, no_currency_symbols=True,\n",
    "                              no_punct=True, no_accents=True)\n",
    "              for x in df['content'].values]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### MULTI-CLASS CLASSIFICATION MODEL"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "tfv = TfidfVectorizer(ngram_range=(1,2), max_features=50000)\n",
    "X = tfv.fit_transform((clean_text))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "rf = RandomForestClassifier(n_estimators=250, verbose=1)\n",
    "knn = KNeighborsClassifier(n_neighbors=5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "[Parallel(n_jobs=1)]: Using backend SequentialBackend with 1 concurrent workers.\n",
      "[Parallel(n_jobs=1)]: Done 250 out of 250 | elapsed:  1.8min finished\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "KNeighborsClassifier(algorithm='auto', leaf_size=30, metric='minkowski',\n",
       "           metric_params=None, n_jobs=None, n_neighbors=5, p=2,\n",
       "           weights='uniform')"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "rf.fit(X_train, y_train)\n",
    "knn.fit(X_train, y_train)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "[Parallel(n_jobs=1)]: Using backend SequentialBackend with 1 concurrent workers.\n",
      "[Parallel(n_jobs=1)]: Done 250 out of 250 | elapsed:    2.8s finished\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "RF: 0.8846620450606586\n",
      "KNN: 0.8304159445407279\n"
     ]
    }
   ],
   "source": [
    "# Random forest and KNN scores:\n",
    "print('RF:', rf.score(X_test, y_test))\n",
    "print('KNN:', knn.score(X_test, y_test))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "[Parallel(n_jobs=1)]: Using backend SequentialBackend with 1 concurrent workers.\n",
      "[Parallel(n_jobs=1)]: Done 250 out of 250 | elapsed:    2.9s finished\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.83      0.72      0.77      1907\n",
      "           1       0.89      0.98      0.94      8917\n",
      "           2       0.92      0.13      0.23       266\n",
      "           3       0.86      0.08      0.15       450\n",
      "\n",
      "   micro avg       0.88      0.88      0.88     11540\n",
      "   macro avg       0.88      0.48      0.52     11540\n",
      "weighted avg       0.88      0.88      0.86     11540\n",
      "\n",
      "[[1381  523    0    3]\n",
      " [ 154 8757    3    3]\n",
      " [  77  154   35    0]\n",
      " [  54  360    0   36]]\n"
     ]
    }
   ],
   "source": [
    "rf_yhat = rf.predict(X_test)\n",
    "#rf_yhat = knn.predict(X_test)\n",
    "\n",
    "print(classification_report(y_test, rf_yhat))\n",
    "print(confusion_matrix(y_test, rf_yhat))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Preparing data for - Predicted cabin class result"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "y_predict = pd.DataFrame(rf_yhat)\n",
    "y_predict.columns = ['predicted_class']\n",
    "y_predict['predicted_class_desc'] = y_predict['predicted_class'].map({1:'Econ',0:'Business',2:'First',3:'Prem'}) "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Preparing data for - Tested subset dataset"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "y_test = df.loc[y_test.index]\n",
    "y_test.reset_index(inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "result = pd.concat([y_test, y_predict], axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {
    "scrolled": true
   },
   "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>content</th>\n",
       "      <th>cabin_flown</th>\n",
       "      <th>cabin_class</th>\n",
       "      <th>predicted_class</th>\n",
       "      <th>predicted_class_desc</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Travelled round trip SIN-ZRH-GVA. All flights ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>18th January EY0455 SYD-AUH. Flight back to Sy...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Flight to Cape Town from CDG. Booked premium e...</td>\n",
       "      <td>3</td>\n",
       "      <td>Premium Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Nadi-Auckland Economy on old 747. All seats an...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>I have used Eastern Airways for a number of ye...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>TBS-DOH-MXP. Both aircraft's seemed relatively...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>EK005 7th Jan Business class with my wife and ...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Intersky is another airline charging high pric...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Jakarta - Kuala Lumpur - Amsterdam - Helsinki ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>HKG-CDG Flight 185. FA's were particularly fri...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>My wife and I travelled business class Sydney ...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>PRG-CGN-PRG. Check-in at PRG okay. Onboard you...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Heathrow to Melbourne in early December and ma...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>GLA to SOU - Perfect flight took off on time l...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Very short flight from Barcelona to Palma (FR ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Crew unfriendly and either don't say anything ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Upon our arrival to Rome from JFK on May 7th w...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>9W 121 London to Delhi. Flight was scheduled t...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>I was positively surprised when I recently in ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Island Air got us safely between Honolulu and ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>Heathrow to LA in Economy. This was my first a...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>Incheon to London Heathrow last month and it w...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>Flew HAN-SGN on an old but clean B777. Flight ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>On July 1 2014 travelled with family from JFK ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>BA12 15th August 2014. In my mind that when I ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>You may think it is a cheap flight - but they ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>Internal flight Daegu - Incheon on which my wi...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>HKG-KUL. The plane is embarrassingly old. My s...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>Manchester to Dalaman Turkey and return June/J...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>I flew Ethiopian Airlines as they are a part o...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>FRA-LCY-FRA on excellent BA Cityflyer - what a...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>Hong Kong - Seoul - Los Angeles return. All fl...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>I had booked a trip to Hong Kong through Fligh...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>LGW to Montego Bay. Premium Cabin The crew wer...</td>\n",
       "      <td>3</td>\n",
       "      <td>Premium Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>I recently has 2 horrible experiences with thi...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>IAD-ORY return. Dulles lounge was satisfactory...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>MEL-JFK over three separate legs. Encountered ...</td>\n",
       "      <td>2</td>\n",
       "      <td>First Class</td>\n",
       "      <td>2</td>\n",
       "      <td>First</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>IST-Paris CDG Business Class A330-200. Boardin...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>Flew from Bahrain to Delhi. Flight took off an...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>Flew LAX-SFO-ORD-LAX last week in First. Had a...</td>\n",
       "      <td>2</td>\n",
       "      <td>First Class</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>QF 98 HKG-BNE on new interior A330. My first f...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>Warsaw-Copenhagen quick and good service. As a...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>My wife and I travelled from CPH to EWR via KE...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>Delayed 9 hours 21 December 2014 from Kilimanj...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>Bellingham to Vegas got to airport on time but...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>LHR-BOM-DEL-LHR. Business. Outbound 777-300ER....</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>Flew round trip from Denver to Frankfurt in Ma...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>I read past reviews and learned that you need ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>We traveled home from Las Vegas on Tuesday Jul...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>Colombo-Muscat-Paris in Business. First leg on...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50</th>\n",
       "      <td>Ground staff for Sydney to Melbourne flight ve...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>51</th>\n",
       "      <td>I travelled Manchester - Sydney return on 777-...</td>\n",
       "      <td>0</td>\n",
       "      <td>Business Class</td>\n",
       "      <td>0</td>\n",
       "      <td>Business</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>52</th>\n",
       "      <td>W63702 and 3701 5th Feb to 12th Feb 2014. I tr...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>53</th>\n",
       "      <td>My husband and I flew from Charlotte NC to Chi...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>54</th>\n",
       "      <td>KUL-ALA-AKX return. First flight from KUL-ALA ...</td>\n",
       "      <td>1</td>\n",
       "      <td>Economy</td>\n",
       "      <td>1</td>\n",
       "      <td>Econ</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                              content  cabin_flown  \\\n",
       "0   Travelled round trip SIN-ZRH-GVA. All flights ...            1   \n",
       "1   18th January EY0455 SYD-AUH. Flight back to Sy...            1   \n",
       "2   Flight to Cape Town from CDG. Booked premium e...            3   \n",
       "3   Nadi-Auckland Economy on old 747. All seats an...            1   \n",
       "4   I have used Eastern Airways for a number of ye...            1   \n",
       "5   TBS-DOH-MXP. Both aircraft's seemed relatively...            1   \n",
       "6   EK005 7th Jan Business class with my wife and ...            0   \n",
       "7   Intersky is another airline charging high pric...            1   \n",
       "8   Jakarta - Kuala Lumpur - Amsterdam - Helsinki ...            1   \n",
       "9   HKG-CDG Flight 185. FA's were particularly fri...            0   \n",
       "10  My wife and I travelled business class Sydney ...            0   \n",
       "11  PRG-CGN-PRG. Check-in at PRG okay. Onboard you...            1   \n",
       "12  Heathrow to Melbourne in early December and ma...            0   \n",
       "13  GLA to SOU - Perfect flight took off on time l...            1   \n",
       "14  Very short flight from Barcelona to Palma (FR ...            1   \n",
       "15  Crew unfriendly and either don't say anything ...            1   \n",
       "16  Upon our arrival to Rome from JFK on May 7th w...            1   \n",
       "17  9W 121 London to Delhi. Flight was scheduled t...            1   \n",
       "18  I was positively surprised when I recently in ...            1   \n",
       "19  Island Air got us safely between Honolulu and ...            1   \n",
       "20  Heathrow to LA in Economy. This was my first a...            1   \n",
       "21  Incheon to London Heathrow last month and it w...            1   \n",
       "22  Flew HAN-SGN on an old but clean B777. Flight ...            1   \n",
       "23  On July 1 2014 travelled with family from JFK ...            1   \n",
       "24  BA12 15th August 2014. In my mind that when I ...            1   \n",
       "25  You may think it is a cheap flight - but they ...            1   \n",
       "26  Internal flight Daegu - Incheon on which my wi...            1   \n",
       "27  HKG-KUL. The plane is embarrassingly old. My s...            0   \n",
       "28  Manchester to Dalaman Turkey and return June/J...            1   \n",
       "29  I flew Ethiopian Airlines as they are a part o...            1   \n",
       "30  FRA-LCY-FRA on excellent BA Cityflyer - what a...            0   \n",
       "31  Hong Kong - Seoul - Los Angeles return. All fl...            1   \n",
       "32  I had booked a trip to Hong Kong through Fligh...            1   \n",
       "33  LGW to Montego Bay. Premium Cabin The crew wer...            3   \n",
       "34  I recently has 2 horrible experiences with thi...            1   \n",
       "35  IAD-ORY return. Dulles lounge was satisfactory...            0   \n",
       "36  MEL-JFK over three separate legs. Encountered ...            2   \n",
       "37  IST-Paris CDG Business Class A330-200. Boardin...            0   \n",
       "38  Flew from Bahrain to Delhi. Flight took off an...            1   \n",
       "39  Flew LAX-SFO-ORD-LAX last week in First. Had a...            2   \n",
       "40  QF 98 HKG-BNE on new interior A330. My first f...            1   \n",
       "41  Warsaw-Copenhagen quick and good service. As a...            1   \n",
       "42  My wife and I travelled from CPH to EWR via KE...            0   \n",
       "43  Delayed 9 hours 21 December 2014 from Kilimanj...            0   \n",
       "44  Bellingham to Vegas got to airport on time but...            1   \n",
       "45  LHR-BOM-DEL-LHR. Business. Outbound 777-300ER....            0   \n",
       "46  Flew round trip from Denver to Frankfurt in Ma...            1   \n",
       "47  I read past reviews and learned that you need ...            1   \n",
       "48  We traveled home from Las Vegas on Tuesday Jul...            1   \n",
       "49  Colombo-Muscat-Paris in Business. First leg on...            0   \n",
       "50  Ground staff for Sydney to Melbourne flight ve...            1   \n",
       "51  I travelled Manchester - Sydney return on 777-...            0   \n",
       "52  W63702 and 3701 5th Feb to 12th Feb 2014. I tr...            1   \n",
       "53  My husband and I flew from Charlotte NC to Chi...            1   \n",
       "54  KUL-ALA-AKX return. First flight from KUL-ALA ...            1   \n",
       "\n",
       "        cabin_class  predicted_class predicted_class_desc  \n",
       "0           Economy                1                 Econ  \n",
       "1           Economy                1                 Econ  \n",
       "2   Premium Economy                1                 Econ  \n",
       "3           Economy                1                 Econ  \n",
       "4           Economy                1                 Econ  \n",
       "5           Economy                1                 Econ  \n",
       "6    Business Class                0             Business  \n",
       "7           Economy                1                 Econ  \n",
       "8           Economy                1                 Econ  \n",
       "9    Business Class                0             Business  \n",
       "10   Business Class                0             Business  \n",
       "11          Economy                1                 Econ  \n",
       "12   Business Class                0             Business  \n",
       "13          Economy                1                 Econ  \n",
       "14          Economy                1                 Econ  \n",
       "15          Economy                1                 Econ  \n",
       "16          Economy                1                 Econ  \n",
       "17          Economy                1                 Econ  \n",
       "18          Economy                1                 Econ  \n",
       "19          Economy                1                 Econ  \n",
       "20          Economy                1                 Econ  \n",
       "21          Economy                1                 Econ  \n",
       "22          Economy                1                 Econ  \n",
       "23          Economy                1                 Econ  \n",
       "24          Economy                1                 Econ  \n",
       "25          Economy                1                 Econ  \n",
       "26          Economy                1                 Econ  \n",
       "27   Business Class                1                 Econ  \n",
       "28          Economy                1                 Econ  \n",
       "29          Economy                1                 Econ  \n",
       "30   Business Class                0             Business  \n",
       "31          Economy                1                 Econ  \n",
       "32          Economy                1                 Econ  \n",
       "33  Premium Economy                1                 Econ  \n",
       "34          Economy                1                 Econ  \n",
       "35   Business Class                0             Business  \n",
       "36      First Class                2                First  \n",
       "37   Business Class                1                 Econ  \n",
       "38          Economy                1                 Econ  \n",
       "39      First Class                1                 Econ  \n",
       "40          Economy                1                 Econ  \n",
       "41          Economy                0             Business  \n",
       "42   Business Class                1                 Econ  \n",
       "43   Business Class                1                 Econ  \n",
       "44          Economy                1                 Econ  \n",
       "45   Business Class                0             Business  \n",
       "46          Economy                1                 Econ  \n",
       "47          Economy                1                 Econ  \n",
       "48          Economy                1                 Econ  \n",
       "49   Business Class                0             Business  \n",
       "50          Economy                1                 Econ  \n",
       "51   Business Class                0             Business  \n",
       "52          Economy                1                 Econ  \n",
       "53          Economy                1                 Econ  \n",
       "54          Economy                1                 Econ  "
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "result[['content','cabin_flown','cabin_class','predicted_class','predicted_class_desc']].head(55)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}