Predict UEFA Champion League quarter final .ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 37,
   "id": "8cebfad9",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Add the libraries\n",
    "import pandas as pd\n",
    "import re\n",
    "import os\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "id": "547c2890",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Teams</th>\n",
       "      <th>elo rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Man City</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Bayern</td>\n",
       "      <td>2347</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>2287</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Barcelona</td>\n",
       "      <td>2260</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Napoli</td>\n",
       "      <td>2252</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Arsenal FC</td>\n",
       "      <td>2212</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Benfica</td>\n",
       "      <td>2196</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Liverpool</td>\n",
       "      <td>2195</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Manchester United</td>\n",
       "      <td>2189</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Inter</td>\n",
       "      <td>2155</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Paris</td>\n",
       "      <td>2150</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Porto</td>\n",
       "      <td>2150</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Dortmund</td>\n",
       "      <td>2141</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Atletico</td>\n",
       "      <td>2140</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Juventus</td>\n",
       "      <td>2129</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Leipzig</td>\n",
       "      <td>2121</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Chelsea</td>\n",
       "      <td>2115</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>Brighton &amp; Hove Albion</td>\n",
       "      <td>2107</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>Milan</td>\n",
       "      <td>2105</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Newcastle United</td>\n",
       "      <td>2103</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>Tottenham</td>\n",
       "      <td>2081</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>Leverkusen</td>\n",
       "      <td>2076</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>Salzburg</td>\n",
       "      <td>2072</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>Real Betis</td>\n",
       "      <td>2063</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>Villarreal CF</td>\n",
       "      <td>2059</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>FC Union Berlin</td>\n",
       "      <td>2058</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>Sporting CP</td>\n",
       "      <td>2056</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>Marseille</td>\n",
       "      <td>2055</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>Feyenoord</td>\n",
       "      <td>2055</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>SC Freiburg</td>\n",
       "      <td>2052</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>AS Roma</td>\n",
       "      <td>2048</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>Ajax</td>\n",
       "      <td>2043</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>Real Sociedad</td>\n",
       "      <td>2043</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>PSV Eindhoven</td>\n",
       "      <td>2039</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>Lazio Roma</td>\n",
       "      <td>2038</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>Frankfurt</td>\n",
       "      <td>2035</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>Celtic</td>\n",
       "      <td>2034</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>Lille OSC</td>\n",
       "      <td>2032</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>Atalanta</td>\n",
       "      <td>2032</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>Aston Villa</td>\n",
       "      <td>2028</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>ACF Fiorentina</td>\n",
       "      <td>2027</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>Brentford FC</td>\n",
       "      <td>2026</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>Zenit St. Petersburg</td>\n",
       "      <td>2025</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>RC Lens</td>\n",
       "      <td>2023</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>AS Monaco</td>\n",
       "      <td>2021</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>Athletic Bilbao</td>\n",
       "      <td>2016</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>West Ham United</td>\n",
       "      <td>2006</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>VfL Wolfsburg</td>\n",
       "      <td>2004</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>Stade Rennes</td>\n",
       "      <td>1996</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>Union Saint Gilloise</td>\n",
       "      <td>1994</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                      Teams  elo rating\n",
       "0                  Man City        2352\n",
       "1                    Bayern        2347\n",
       "2               Real Madrid        2287\n",
       "3                 Barcelona        2260\n",
       "4                    Napoli        2252\n",
       "5               Arsenal FC         2212\n",
       "6                   Benfica        2196\n",
       "7                 Liverpool        2195\n",
       "8        Manchester United         2189\n",
       "9                     Inter        2155\n",
       "10                    Paris        2150\n",
       "11                    Porto        2150\n",
       "12                 Dortmund        2141\n",
       "13                 Atletico        2140\n",
       "14                 Juventus        2129\n",
       "15                  Leipzig        2121\n",
       "16                  Chelsea        2115\n",
       "17  Brighton & Hove Albion         2107\n",
       "18                    Milan        2105\n",
       "19        Newcastle United         2103\n",
       "20                Tottenham        2081\n",
       "21               Leverkusen        2076\n",
       "22                 Salzburg        2072\n",
       "23              Real Betis         2063\n",
       "24           Villarreal CF         2059\n",
       "25         FC Union Berlin         2058\n",
       "26             Sporting CP         2056\n",
       "27                Marseille        2055\n",
       "28               Feyenoord         2055\n",
       "29             SC Freiburg         2052\n",
       "30                 AS Roma         2048\n",
       "31                     Ajax        2043\n",
       "32           Real Sociedad         2043\n",
       "33           PSV Eindhoven         2039\n",
       "34              Lazio Roma         2038\n",
       "35                Frankfurt        2035\n",
       "36                   Celtic        2034\n",
       "37               Lille OSC         2032\n",
       "38                Atalanta         2032\n",
       "39             Aston Villa         2028\n",
       "40          ACF Fiorentina         2027\n",
       "41            Brentford FC         2026\n",
       "42    Zenit St. Petersburg         2025\n",
       "43                 RC Lens         2023\n",
       "44               AS Monaco         2021\n",
       "45         Athletic Bilbao         2016\n",
       "46         West Ham United         2006\n",
       "47           VfL Wolfsburg         2004\n",
       "48            Stade Rennes         1996\n",
       "49    Union Saint Gilloise         1994"
      ]
     },
     "execution_count": 38,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "elo_ratings = pd.read_csv('D:/STUDY/STUDY/PYTHON/Predict winner/elo rating.csv')\n",
    "elo_ratings"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "id": "ad900403",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Teams</th>\n",
       "      <th>Attacks</th>\n",
       "      <th>Defense</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Man City</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>459</td>\n",
       "      <td>294</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Liverpool</td>\n",
       "      <td>446</td>\n",
       "      <td>322</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Chelsea</td>\n",
       "      <td>430</td>\n",
       "      <td>365</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Bayern</td>\n",
       "      <td>410</td>\n",
       "      <td>351</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Barcelona</td>\n",
       "      <td>398</td>\n",
       "      <td>231</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Paris</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Benfica</td>\n",
       "      <td>367</td>\n",
       "      <td>327</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Napoli</td>\n",
       "      <td>361</td>\n",
       "      <td>344</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Tottenham</td>\n",
       "      <td>349</td>\n",
       "      <td>326</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Inter</td>\n",
       "      <td>340</td>\n",
       "      <td>287</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Marseille</td>\n",
       "      <td>336</td>\n",
       "      <td>243</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Porto</td>\n",
       "      <td>314</td>\n",
       "      <td>329</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Milan</td>\n",
       "      <td>312</td>\n",
       "      <td>312</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Atletico</td>\n",
       "      <td>309</td>\n",
       "      <td>236</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Dortmund</td>\n",
       "      <td>302</td>\n",
       "      <td>323</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Leipzig</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>Juventus</td>\n",
       "      <td>283</td>\n",
       "      <td>220</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>Leverkusen</td>\n",
       "      <td>278</td>\n",
       "      <td>266</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Club Brugge</td>\n",
       "      <td>275</td>\n",
       "      <td>295</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>Salzburg</td>\n",
       "      <td>249</td>\n",
       "      <td>272</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>Ajax</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>Celtic</td>\n",
       "      <td>246</td>\n",
       "      <td>218</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>Frankfurt</td>\n",
       "      <td>243</td>\n",
       "      <td>376</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>M. Haifa</td>\n",
       "      <td>222</td>\n",
       "      <td>244</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>Sevilla</td>\n",
       "      <td>201</td>\n",
       "      <td>207</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>Plzen</td>\n",
       "      <td>184</td>\n",
       "      <td>223</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>Dinamo Zagreb</td>\n",
       "      <td>177</td>\n",
       "      <td>236</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>Copenhagen</td>\n",
       "      <td>170</td>\n",
       "      <td>221</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>Rangers</td>\n",
       "      <td>166</td>\n",
       "      <td>219</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>Sporting CP</td>\n",
       "      <td>158</td>\n",
       "      <td>226</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>Shakhtar Donetsk</td>\n",
       "      <td>133</td>\n",
       "      <td>234</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Teams  Attacks  Defense\n",
       "0           Man City      529      284\n",
       "1        Real Madrid      459      294\n",
       "2          Liverpool      446      322\n",
       "3            Chelsea      430      365\n",
       "4             Bayern      410      351\n",
       "5          Barcelona      398      231\n",
       "6              Paris      392      333\n",
       "7            Benfica      367      327\n",
       "8             Napoli      361      344\n",
       "9          Tottenham      349      326\n",
       "10             Inter      340      287\n",
       "11         Marseille      336      243\n",
       "12             Porto      314      329\n",
       "13             Milan      312      312\n",
       "14          Atletico      309      236\n",
       "15          Dortmund      302      323\n",
       "16           Leipzig      293      317\n",
       "17          Juventus      283      220\n",
       "18        Leverkusen      278      266\n",
       "19       Club Brugge      275      295\n",
       "20          Salzburg      249      272\n",
       "21              Ajax      247      246\n",
       "22            Celtic      246      218\n",
       "23         Frankfurt      243      376\n",
       "24          M. Haifa      222      244\n",
       "25           Sevilla      201      207\n",
       "26             Plzen      184      223\n",
       "27     Dinamo Zagreb      177      236\n",
       "28        Copenhagen      170      221\n",
       "29           Rangers      166      219\n",
       "30       Sporting CP      158      226\n",
       "31  Shakhtar Donetsk      133      234"
      ]
     },
     "execution_count": 39,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "team_stats = pd.read_csv('D:/STUDY/STUDY/PYTHON/Predict winner/team stats.csv')\n",
    "team_stats"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "id": "5d162ebf",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>DATE</th>\n",
       "      <th>SEASON</th>\n",
       "      <th>HOME_TEAM</th>\n",
       "      <th>AWAY_TEAM</th>\n",
       "      <th>HOME_TEAM_SCORE</th>\n",
       "      <th>AWAY_TEAM_SCORE</th>\n",
       "      <th>Winner</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>9/15/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>Man City</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>9/15/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Club Brugge</td>\n",
       "      <td>Paris</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>Draw</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>9/28/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Paris</td>\n",
       "      <td>Man City</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>Paris</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>9/28/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>Club Brugge</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>Club Brugge</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>10/19/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Club Brugge</td>\n",
       "      <td>Man City</td>\n",
       "      <td>1</td>\n",
       "      <td>5</td>\n",
       "      <td>Man City</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>739</th>\n",
       "      <td>5/2/2017</td>\n",
       "      <td>2016-2017</td>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>Atletico</td>\n",
       "      <td>3</td>\n",
       "      <td>0</td>\n",
       "      <td>Real Madrid</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>740</th>\n",
       "      <td>5/3/2017</td>\n",
       "      <td>2016-2017</td>\n",
       "      <td>AS Monaco</td>\n",
       "      <td>Juventus</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>Juventus</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>741</th>\n",
       "      <td>5/9/2017</td>\n",
       "      <td>2016-2017</td>\n",
       "      <td>Juventus</td>\n",
       "      <td>AS Monaco</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>Juventus</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>742</th>\n",
       "      <td>5/10/2017</td>\n",
       "      <td>2016-2017</td>\n",
       "      <td>Atletico</td>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>Atletico</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>743</th>\n",
       "      <td>6/3/2017</td>\n",
       "      <td>2016-2017</td>\n",
       "      <td>Juventus</td>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>1</td>\n",
       "      <td>4</td>\n",
       "      <td>Real Madrid</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>744 rows × 7 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "           DATE     SEASON    HOME_TEAM    AWAY_TEAM  HOME_TEAM_SCORE  \\\n",
       "0     9/15/2021  2021-2022     Man City      Leipzig                6   \n",
       "1     9/15/2021  2021-2022  Club Brugge        Paris                1   \n",
       "2     9/28/2021  2021-2022        Paris     Man City                2   \n",
       "3     9/28/2021  2021-2022      Leipzig  Club Brugge                1   \n",
       "4    10/19/2021  2021-2022  Club Brugge     Man City                1   \n",
       "..          ...        ...          ...          ...              ...   \n",
       "739    5/2/2017  2016-2017  Real Madrid     Atletico                3   \n",
       "740    5/3/2017  2016-2017    AS Monaco     Juventus                0   \n",
       "741    5/9/2017  2016-2017     Juventus    AS Monaco                2   \n",
       "742   5/10/2017  2016-2017     Atletico  Real Madrid                2   \n",
       "743    6/3/2017  2016-2017     Juventus  Real Madrid                1   \n",
       "\n",
       "     AWAY_TEAM_SCORE       Winner  \n",
       "0                  3     Man City  \n",
       "1                  1         Draw  \n",
       "2                  0        Paris  \n",
       "3                  2  Club Brugge  \n",
       "4                  5     Man City  \n",
       "..               ...          ...  \n",
       "739                0  Real Madrid  \n",
       "740                2     Juventus  \n",
       "741                1     Juventus  \n",
       "742                1     Atletico  \n",
       "743                4  Real Madrid  \n",
       "\n",
       "[744 rows x 7 columns]"
      ]
     },
     "execution_count": 40,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "matches = pd.read_csv('D:/STUDY/STUDY/PYTHON/Predict winner/matches.csv')\n",
    "matches"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 62,
   "id": "8d0e9673",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Teams</th>\n",
       "      <th>Attacks</th>\n",
       "      <th>Defense</th>\n",
       "      <th>elo rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Man City</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Real Madrid</td>\n",
       "      <td>459</td>\n",
       "      <td>294</td>\n",
       "      <td>2287</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Liverpool</td>\n",
       "      <td>446</td>\n",
       "      <td>322</td>\n",
       "      <td>2195</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Chelsea</td>\n",
       "      <td>430</td>\n",
       "      <td>365</td>\n",
       "      <td>2115</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Bayern</td>\n",
       "      <td>410</td>\n",
       "      <td>351</td>\n",
       "      <td>2347</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Barcelona</td>\n",
       "      <td>398</td>\n",
       "      <td>231</td>\n",
       "      <td>2260</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>Paris</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>Benfica</td>\n",
       "      <td>367</td>\n",
       "      <td>327</td>\n",
       "      <td>2196</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>Napoli</td>\n",
       "      <td>361</td>\n",
       "      <td>344</td>\n",
       "      <td>2252</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>Tottenham</td>\n",
       "      <td>349</td>\n",
       "      <td>326</td>\n",
       "      <td>2081</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>Inter</td>\n",
       "      <td>340</td>\n",
       "      <td>287</td>\n",
       "      <td>2155</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>Marseille</td>\n",
       "      <td>336</td>\n",
       "      <td>243</td>\n",
       "      <td>2055</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Porto</td>\n",
       "      <td>314</td>\n",
       "      <td>329</td>\n",
       "      <td>2150</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Milan</td>\n",
       "      <td>312</td>\n",
       "      <td>312</td>\n",
       "      <td>2105</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>Atletico</td>\n",
       "      <td>309</td>\n",
       "      <td>236</td>\n",
       "      <td>2140</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>Dortmund</td>\n",
       "      <td>302</td>\n",
       "      <td>323</td>\n",
       "      <td>2141</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Leipzig</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>Juventus</td>\n",
       "      <td>283</td>\n",
       "      <td>220</td>\n",
       "      <td>2129</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>Leverkusen</td>\n",
       "      <td>278</td>\n",
       "      <td>266</td>\n",
       "      <td>2076</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>Salzburg</td>\n",
       "      <td>249</td>\n",
       "      <td>272</td>\n",
       "      <td>2072</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>Ajax</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>Celtic</td>\n",
       "      <td>246</td>\n",
       "      <td>218</td>\n",
       "      <td>2034</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>Frankfurt</td>\n",
       "      <td>243</td>\n",
       "      <td>376</td>\n",
       "      <td>2035</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "          Teams  Attacks  Defense  elo rating\n",
       "0      Man City      529      284        2352\n",
       "1   Real Madrid      459      294        2287\n",
       "2     Liverpool      446      322        2195\n",
       "3       Chelsea      430      365        2115\n",
       "4        Bayern      410      351        2347\n",
       "5     Barcelona      398      231        2260\n",
       "6         Paris      392      333        2150\n",
       "7       Benfica      367      327        2196\n",
       "8        Napoli      361      344        2252\n",
       "9     Tottenham      349      326        2081\n",
       "10        Inter      340      287        2155\n",
       "11    Marseille      336      243        2055\n",
       "12        Porto      314      329        2150\n",
       "13        Milan      312      312        2105\n",
       "14     Atletico      309      236        2140\n",
       "15     Dortmund      302      323        2141\n",
       "16      Leipzig      293      317        2121\n",
       "17     Juventus      283      220        2129\n",
       "18   Leverkusen      278      266        2076\n",
       "19     Salzburg      249      272        2072\n",
       "20         Ajax      247      246        2043\n",
       "21       Celtic      246      218        2034\n",
       "22    Frankfurt      243      376        2035"
      ]
     },
     "execution_count": 62,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "team_data = pd.merge(team_stats, elo_ratings, left_on = 'Teams', right_on='Teams', how='inner').reset_index(drop=True)\n",
    "team_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "id": "585a2075",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>DATE</th>\n",
       "      <th>SEASON</th>\n",
       "      <th>HOME_TEAM</th>\n",
       "      <th>AWAY_TEAM</th>\n",
       "      <th>HOME_TEAM_SCORE</th>\n",
       "      <th>AWAY_TEAM_SCORE</th>\n",
       "      <th>home_attack</th>\n",
       "      <th>home_defense</th>\n",
       "      <th>home_elo_rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>9/15/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>11/3/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Club Brugge</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>11/24/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Paris</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3/9/2022</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Sporting CP</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>4/5/2022</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Atletico</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "         DATE     SEASON HOME_TEAM    AWAY_TEAM  HOME_TEAM_SCORE  \\\n",
       "0   9/15/2021  2021-2022  Man City      Leipzig                6   \n",
       "1   11/3/2021  2021-2022  Man City  Club Brugge                4   \n",
       "2  11/24/2021  2021-2022  Man City        Paris                2   \n",
       "3    3/9/2022  2021-2022  Man City  Sporting CP                0   \n",
       "4    4/5/2022  2021-2022  Man City     Atletico                1   \n",
       "\n",
       "   AWAY_TEAM_SCORE  home_attack  home_defense  home_elo_rating  \n",
       "0                3          529           284             2352  \n",
       "1                1          529           284             2352  \n",
       "2                1          529           284             2352  \n",
       "3                0          529           284             2352  \n",
       "4                0          529           284             2352  "
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "matches = pd.merge(matches, team_data, left_on = 'HOME_TEAM', right_on = 'Teams', how='inner').reset_index(drop=True)\n",
    "matches.rename(columns = {'Attacks':'home_attack','Defense':'home_defense','elo rating':'home_elo_rating'}, inplace=True)\n",
    "matches.drop(['Teams'], axis=1, inplace=True)\n",
    "matches.drop(['Winner'], axis=1, inplace=True)\n",
    "matches.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "id": "96f53f7e",
   "metadata": {},
   "outputs": [],
   "source": [
    "matches = pd.merge(matches, team_data, left_on = 'AWAY_TEAM', right_on = 'Teams', how='inner').reset_index(drop=True)\n",
    "matches.rename(columns = {'Attacks':'away_attack','Defense':'away_defense','elo rating':'away_elo_rating'}, inplace=True)\n",
    "matches.drop(['Teams'], axis=1, inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "id": "c958d270",
   "metadata": {},
   "outputs": [],
   "source": [
    "matches['result'] = np.where(matches['HOME_TEAM_SCORE'] > matches['AWAY_TEAM_SCORE'], 'W',\n",
    "                    np.where(matches['HOME_TEAM_SCORE'] < matches['AWAY_TEAM_SCORE'], 'L', 'D'))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "id": "dd4ff6bb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "         DATE     SEASON  HOME_TEAM AWAY_TEAM  HOME_TEAM_SCORE  \\\n",
      "0   9/15/2021  2021-2022   Man City   Leipzig                6   \n",
      "1  10/19/2021  2021-2022      Paris   Leipzig                3   \n",
      "2  11/24/2020  2020-2021      Paris   Leipzig                1   \n",
      "3   3/10/2021  2020-2021  Liverpool   Leipzig                2   \n",
      "4   11/1/2017  2017-2018      Porto   Leipzig                3   \n",
      "5   9/17/2019  2019-2020    Benfica   Leipzig                1   \n",
      "6   2/19/2020  2019-2020  Tottenham   Leipzig                0   \n",
      "7  11/24/2021  2021-2022   Man City     Paris                2   \n",
      "8    5/4/2021  2020-2021   Man City     Paris                2   \n",
      "9   11/3/2021  2021-2022    Leipzig     Paris                2   \n",
      "\n",
      "   AWAY_TEAM_SCORE  home_attack  home_defense  home_elo_rating  away_attack  \\\n",
      "0                3          529           284             2352          293   \n",
      "1                2          392           333             2150          293   \n",
      "2                0          392           333             2150          293   \n",
      "3                0          446           322             2195          293   \n",
      "4                1          314           329             2150          293   \n",
      "5                2          367           327             2196          293   \n",
      "6                1          349           326             2081          293   \n",
      "7                1          529           284             2352          392   \n",
      "8                0          529           284             2352          392   \n",
      "9                2          293           317             2121          392   \n",
      "\n",
      "   away_defense  away_elo_rating result  \n",
      "0           317             2121      W  \n",
      "1           317             2121      W  \n",
      "2           317             2121      W  \n",
      "3           317             2121      W  \n",
      "4           317             2121      W  \n",
      "5           317             2121      L  \n",
      "6           317             2121      L  \n",
      "7           333             2150      W  \n",
      "8           333             2150      W  \n",
      "9           333             2150      D  \n"
     ]
    }
   ],
   "source": [
    "print(matches.head(10))   "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "id": "f1461a6d",
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.pyplot as plt\n",
    "import seaborn as sns\n",
    "%matplotlib inline\n",
    "import warnings\n",
    "warnings.filterwarnings('ignore')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "id": "ad29629d",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.cluster import KMeans\n",
    "cluster_data = team_data[['Attacks','Defense','elo rating']]\n",
    "cluster_data['Attacks'] = cluster_data['Attacks'].astype('float')\n",
    "cluster_data['Defense'] = cluster_data['Defense'].astype('float')\n",
    "cluster_data['elo rating'] = cluster_data['elo rating'].astype('float')\n",
    "\n",
    "cluster_data = (cluster_data-cluster_data.min())/(cluster_data.max()-cluster_data.min())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 80,
   "id": "e91bf370",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "          Teams  Attacks  Defense  elo rating  Cluster\n",
      "0      Man City      529      284        2352   Strong\n",
      "1   Real Madrid      459      294        2287   Strong\n",
      "2     Liverpool      446      322        2195   Strong\n",
      "3       Chelsea      430      365        2115  Average\n",
      "4        Bayern      410      351        2347   Strong\n",
      "5     Barcelona      398      231        2260   Strong\n",
      "6         Paris      392      333        2150  Average\n",
      "7       Benfica      367      327        2196  Average\n",
      "8        Napoli      361      344        2252  Average\n",
      "9     Tottenham      349      326        2081  Average\n",
      "10        Inter      340      287        2155  Average\n",
      "11    Marseille      336      243        2055     Weak\n",
      "12        Porto      314      329        2150  Average\n",
      "13        Milan      312      312        2105  Average\n",
      "14     Atletico      309      236        2140     Weak\n",
      "15     Dortmund      302      323        2141  Average\n",
      "16      Leipzig      293      317        2121  Average\n",
      "17     Juventus      283      220        2129     Weak\n",
      "18   Leverkusen      278      266        2076     Weak\n",
      "19     Salzburg      249      272        2072     Weak\n",
      "20         Ajax      247      246        2043     Weak\n",
      "21       Celtic      246      218        2034     Weak\n",
      "22    Frankfurt      243      376        2035  Average\n"
     ]
    }
   ],
   "source": [
    "# Clasify team into 'Weak', 'Average' and 'Strong' categories\n",
    "kmeans = KMeans(n_clusters=3)\n",
    "kmeans.fit(cluster_data)\n",
    "cluster_data['Cluster'] = kmeans.labels_\n",
    "cluster_numbers = cluster_data.groupby(['Cluster']).mean('elo rating').sort_values('elo rating')\n",
    "cluster_order = cluster_numbers.index.unique()\n",
    "\n",
    "team_data['Cluster']=cluster_data['Cluster'].astype('str')\n",
    "team_data['Cluster'].replace(str(cluster_order[0]),'Weak', inplace=True)\n",
    "team_data['Cluster'].replace(str(cluster_order[1]),'Average', inplace=True)\n",
    "team_data['Cluster'].replace(str(cluster_order[2]),'Strong', inplace=True)\n",
    "print(team_data.to_string())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "id": "49b5748a",
   "metadata": {},
   "outputs": [],
   "source": [
    "#the dict() function is called with team_data[['Teams','Cluster']].to_records(index=False) as an argument to create a dictionary \n",
    "# for mapping team names to their respective clusters, \n",
    "# then the resulting dictionary is used to map the team names in the HOME_TEAM and AWAY_TEAM columns to their corresponding clusters.\n",
    "\n",
    "matches['home_cluster'] = matches['HOME_TEAM'].map(dict(team_data[['Teams','Cluster']].to_records(index=False)))\n",
    "matches['away_cluster'] = matches['AWAY_TEAM'].map(dict(team_data[['Teams','Cluster']].to_records(index=False)))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "id": "1a4cb457",
   "metadata": {},
   "outputs": [],
   "source": [
    "data = matches.dropna()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "id": "9278563f",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>DATE</th>\n",
       "      <th>SEASON</th>\n",
       "      <th>HOME_TEAM</th>\n",
       "      <th>AWAY_TEAM</th>\n",
       "      <th>HOME_TEAM_SCORE</th>\n",
       "      <th>AWAY_TEAM_SCORE</th>\n",
       "      <th>home_attack</th>\n",
       "      <th>home_defense</th>\n",
       "      <th>home_elo_rating</th>\n",
       "      <th>away_attack</th>\n",
       "      <th>away_defense</th>\n",
       "      <th>away_elo_rating</th>\n",
       "      <th>result</th>\n",
       "      <th>home_cluster</th>\n",
       "      <th>away_cluster</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>9/15/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Man City</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10/19/2021</td>\n",
       "      <td>2021-2022</td>\n",
       "      <td>Paris</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>11/24/2020</td>\n",
       "      <td>2020-2021</td>\n",
       "      <td>Paris</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3/10/2021</td>\n",
       "      <td>2020-2021</td>\n",
       "      <td>Liverpool</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>446</td>\n",
       "      <td>322</td>\n",
       "      <td>2195</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>11/1/2017</td>\n",
       "      <td>2017-2018</td>\n",
       "      <td>Porto</td>\n",
       "      <td>Leipzig</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "      <td>314</td>\n",
       "      <td>329</td>\n",
       "      <td>2150</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>230</th>\n",
       "      <td>11/7/2018</td>\n",
       "      <td>2018-2019</td>\n",
       "      <td>Benfica</td>\n",
       "      <td>Ajax</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>367</td>\n",
       "      <td>327</td>\n",
       "      <td>2196</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>231</th>\n",
       "      <td>10/2/2018</td>\n",
       "      <td>2018-2019</td>\n",
       "      <td>Bayern</td>\n",
       "      <td>Ajax</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>410</td>\n",
       "      <td>351</td>\n",
       "      <td>2347</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>2</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>232</th>\n",
       "      <td>11/5/2019</td>\n",
       "      <td>2019-2020</td>\n",
       "      <td>Chelsea</td>\n",
       "      <td>Ajax</td>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>430</td>\n",
       "      <td>365</td>\n",
       "      <td>2115</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>2</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>233</th>\n",
       "      <td>4/16/2019</td>\n",
       "      <td>2018-2019</td>\n",
       "      <td>Juventus</td>\n",
       "      <td>Ajax</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>283</td>\n",
       "      <td>220</td>\n",
       "      <td>2129</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>234</th>\n",
       "      <td>4/30/2019</td>\n",
       "      <td>2018-2019</td>\n",
       "      <td>Tottenham</td>\n",
       "      <td>Ajax</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>349</td>\n",
       "      <td>326</td>\n",
       "      <td>2081</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>235 rows × 15 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "           DATE     SEASON  HOME_TEAM AWAY_TEAM  HOME_TEAM_SCORE  \\\n",
       "0     9/15/2021  2021-2022   Man City   Leipzig                6   \n",
       "1    10/19/2021  2021-2022      Paris   Leipzig                3   \n",
       "2    11/24/2020  2020-2021      Paris   Leipzig                1   \n",
       "3     3/10/2021  2020-2021  Liverpool   Leipzig                2   \n",
       "4     11/1/2017  2017-2018      Porto   Leipzig                3   \n",
       "..          ...        ...        ...       ...              ...   \n",
       "230   11/7/2018  2018-2019    Benfica      Ajax                1   \n",
       "231   10/2/2018  2018-2019     Bayern      Ajax                1   \n",
       "232   11/5/2019  2019-2020    Chelsea      Ajax                4   \n",
       "233   4/16/2019  2018-2019   Juventus      Ajax                1   \n",
       "234   4/30/2019  2018-2019  Tottenham      Ajax                0   \n",
       "\n",
       "     AWAY_TEAM_SCORE  home_attack  home_defense  home_elo_rating  away_attack  \\\n",
       "0                  3          529           284             2352          293   \n",
       "1                  2          392           333             2150          293   \n",
       "2                  0          392           333             2150          293   \n",
       "3                  0          446           322             2195          293   \n",
       "4                  1          314           329             2150          293   \n",
       "..               ...          ...           ...              ...          ...   \n",
       "230                1          367           327             2196          247   \n",
       "231                1          410           351             2347          247   \n",
       "232                4          430           365             2115          247   \n",
       "233                2          283           220             2129          247   \n",
       "234                1          349           326             2081          247   \n",
       "\n",
       "     away_defense  away_elo_rating  result  home_cluster  away_cluster  \n",
       "0             317             2121       3             3             2  \n",
       "1             317             2121       3             2             2  \n",
       "2             317             2121       3             2             2  \n",
       "3             317             2121       3             3             2  \n",
       "4             317             2121       3             2             2  \n",
       "..            ...              ...     ...           ...           ...  \n",
       "230           246             2043       2             2             1  \n",
       "231           246             2043       2             3             1  \n",
       "232           246             2043       2             2             1  \n",
       "233           246             2043       1             1             1  \n",
       "234           246             2043       1             2             1  \n",
       "\n",
       "[235 rows x 15 columns]"
      ]
     },
     "execution_count": 53,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Since sklearn work with numeric data -> convert categories into numeric categories\n",
    "data['home_cluster'].replace('Strong',3,inplace=True)\n",
    "data['home_cluster'].replace('Average',2,inplace=True)\n",
    "data['home_cluster'].replace('Weak',1,inplace=True)\n",
    "data['away_cluster'].replace('Strong',3,inplace=True)\n",
    "data['away_cluster'].replace('Average',2,inplace=True)\n",
    "data['away_cluster'].replace('Weak',1,inplace=True)\n",
    "data['result'].replace('W',3,inplace=True)\n",
    "data['result'].replace('D',2,inplace=True)\n",
    "data['result'].replace('L',1,inplace=True)\n",
    "data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 54,
   "id": "f9f4b204",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>HOME_TEAM_SCORE</th>\n",
       "      <th>AWAY_TEAM_SCORE</th>\n",
       "      <th>result</th>\n",
       "      <th>home_attack</th>\n",
       "      <th>home_defense</th>\n",
       "      <th>home_elo_rating</th>\n",
       "      <th>home_cluster</th>\n",
       "      <th>away_attack</th>\n",
       "      <th>away_defense</th>\n",
       "      <th>away_elo_rating</th>\n",
       "      <th>away_cluster</th>\n",
       "      <th>attack_effective</th>\n",
       "      <th>defense_effective</th>\n",
       "      <th>elo_rating_diff</th>\n",
       "      <th>team_strength_diff</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "      <td>3</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>212</td>\n",
       "      <td>-9</td>\n",
       "      <td>231</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>3</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>75</td>\n",
       "      <td>40</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>75</td>\n",
       "      <td>40</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>446</td>\n",
       "      <td>322</td>\n",
       "      <td>2195</td>\n",
       "      <td>3</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>129</td>\n",
       "      <td>29</td>\n",
       "      <td>74</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>314</td>\n",
       "      <td>329</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>-3</td>\n",
       "      <td>36</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>230</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>367</td>\n",
       "      <td>327</td>\n",
       "      <td>2196</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>121</td>\n",
       "      <td>80</td>\n",
       "      <td>153</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>231</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>410</td>\n",
       "      <td>351</td>\n",
       "      <td>2347</td>\n",
       "      <td>3</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>164</td>\n",
       "      <td>104</td>\n",
       "      <td>304</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>232</th>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>430</td>\n",
       "      <td>365</td>\n",
       "      <td>2115</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>184</td>\n",
       "      <td>118</td>\n",
       "      <td>72</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>233</th>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>283</td>\n",
       "      <td>220</td>\n",
       "      <td>2129</td>\n",
       "      <td>1</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>37</td>\n",
       "      <td>-27</td>\n",
       "      <td>86</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>234</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>349</td>\n",
       "      <td>326</td>\n",
       "      <td>2081</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>103</td>\n",
       "      <td>79</td>\n",
       "      <td>38</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>235 rows × 15 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     HOME_TEAM_SCORE  AWAY_TEAM_SCORE  result  home_attack  home_defense  \\\n",
       "0                  6                3       3          529           284   \n",
       "1                  3                2       3          392           333   \n",
       "2                  1                0       3          392           333   \n",
       "3                  2                0       3          446           322   \n",
       "4                  3                1       3          314           329   \n",
       "..               ...              ...     ...          ...           ...   \n",
       "230                1                1       2          367           327   \n",
       "231                1                1       2          410           351   \n",
       "232                4                4       2          430           365   \n",
       "233                1                2       1          283           220   \n",
       "234                0                1       1          349           326   \n",
       "\n",
       "     home_elo_rating  home_cluster  away_attack  away_defense  \\\n",
       "0               2352             3          293           317   \n",
       "1               2150             2          293           317   \n",
       "2               2150             2          293           317   \n",
       "3               2195             3          293           317   \n",
       "4               2150             2          293           317   \n",
       "..               ...           ...          ...           ...   \n",
       "230             2196             2          247           246   \n",
       "231             2347             3          247           246   \n",
       "232             2115             2          247           246   \n",
       "233             2129             1          247           246   \n",
       "234             2081             2          247           246   \n",
       "\n",
       "     away_elo_rating  away_cluster  attack_effective  defense_effective  \\\n",
       "0               2121             2               212                 -9   \n",
       "1               2121             2                75                 40   \n",
       "2               2121             2                75                 40   \n",
       "3               2121             2               129                 29   \n",
       "4               2121             2                -3                 36   \n",
       "..               ...           ...               ...                ...   \n",
       "230             2043             1               121                 80   \n",
       "231             2043             1               164                104   \n",
       "232             2043             1               184                118   \n",
       "233             2043             1                37                -27   \n",
       "234             2043             1               103                 79   \n",
       "\n",
       "     elo_rating_diff  team_strength_diff  \n",
       "0                231                   1  \n",
       "1                 29                   0  \n",
       "2                 29                   0  \n",
       "3                 74                   1  \n",
       "4                 29                   0  \n",
       "..               ...                 ...  \n",
       "230              153                   1  \n",
       "231              304                   2  \n",
       "232               72                   1  \n",
       "233               86                   0  \n",
       "234               38                   1  \n",
       "\n",
       "[235 rows x 15 columns]"
      ]
     },
     "execution_count": 54,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data = data[['HOME_TEAM_SCORE', 'AWAY_TEAM_SCORE', 'result',\n",
    "             'home_attack', 'home_defense', 'home_elo_rating', 'home_cluster',\n",
    "             'away_attack', 'away_defense', 'away_elo_rating', 'away_cluster']]\n",
    "\n",
    "data = data.apply(pd.to_numeric)\n",
    "data['attack_effective'] = data['home_attack']-data['away_defense']\n",
    "data['defense_effective'] = data['home_defense']-data['away_attack']\n",
    "data['elo_rating_diff'] = data['home_elo_rating']-data['away_elo_rating']\n",
    "data['team_strength_diff'] = data['home_cluster']-data['away_cluster']\n",
    "\n",
    "data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 55,
   "id": "5a17c69e",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.model_selection import train_test_split\n",
    "X = data[['home_attack','home_defense', 'home_elo_rating', 'home_cluster',\n",
    "          'away_attack','away_defense', 'away_elo_rating', 'away_cluster', \n",
    "          'attack_effective','defense_effective', 'elo_rating_diff','team_strength_diff']]\n",
    "\n",
    "X = (X-X.min())/(X.max()-X.min())\n",
    "y = data[['HOME_TEAM_SCORE', 'AWAY_TEAM_SCORE', 'result']]\n",
    "\n",
    "X_train, X_test, y_train, y_test = train_test_split(X,y,test_size=0.1, shuffle=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 56,
   "id": "45deba2d",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.ensemble import RandomForestClassifier\n",
    "\n",
    "clf = RandomForestClassifier(n_estimators=100, max_depth=5)\n",
    "clf.fit(X_train, y_train)\n",
    "\n",
    "y_pred = clf.predict(X_test)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 57,
   "id": "e5e59d8a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Accuracy:  0.4166666666666667\n"
     ]
    }
   ],
   "source": [
    "from sklearn import metrics\n",
    "accuracy = metrics.accuracy_score(y_test['result'], y_pred[:,2])\n",
    "print(\"Accuracy: \",accuracy)\n",
    "     "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 58,
   "id": "30fa3615",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.metrics import confusion_matrix"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 59,
   "id": "253c5d91",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "elo_rating_diff       0.149461\n",
      "defense_effective     0.132121\n",
      "attack_effective      0.127169\n",
      "home_elo_rating       0.095504\n",
      "away_attack           0.088258\n",
      "home_attack           0.087367\n",
      "away_defense          0.083797\n",
      "home_defense          0.070314\n",
      "away_elo_rating       0.066941\n",
      "team_strength_diff    0.062197\n",
      "away_cluster          0.019065\n",
      "home_cluster          0.017806\n",
      "dtype: float64\n"
     ]
    }
   ],
   "source": [
    "feature_imp = pd.Series(clf.feature_importances_, index=X.columns).sort_values(ascending=False)\n",
    "print(feature_imp)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 63,
   "id": "1f76aaec",
   "metadata": {},
   "outputs": [],
   "source": [
    "X_train = X_train[['attack_effective', 'defense_effective','elo_rating_diff']]\n",
    "X_test = X_test[['attack_effective', 'defense_effective','elo_rating_diff']]\n",
    "\n",
    "clf = RandomForestClassifier(n_estimators=100, max_depth=5)\n",
    "clf.fit(X_train, y_train)\n",
    "\n",
    "y_pred = clf.predict(X_test)\n",
    "\n",
    "accuracy = metrics.accuracy_score(y_test['result'], y_pred[:,2])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 67,
   "id": "1e446cd1",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>HOME_TEAM_SCORE</th>\n",
       "      <th>AWAY_TEAM_SCORE</th>\n",
       "      <th>result</th>\n",
       "      <th>home_attack</th>\n",
       "      <th>home_defense</th>\n",
       "      <th>home_elo_rating</th>\n",
       "      <th>home_cluster</th>\n",
       "      <th>away_attack</th>\n",
       "      <th>away_defense</th>\n",
       "      <th>away_elo_rating</th>\n",
       "      <th>away_cluster</th>\n",
       "      <th>attack_effective</th>\n",
       "      <th>defense_effective</th>\n",
       "      <th>elo_rating_diff</th>\n",
       "      <th>team_strength_diff</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>6</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "      <td>529</td>\n",
       "      <td>284</td>\n",
       "      <td>2352</td>\n",
       "      <td>3</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>212</td>\n",
       "      <td>-9</td>\n",
       "      <td>231</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "      <td>3</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>75</td>\n",
       "      <td>40</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>392</td>\n",
       "      <td>333</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>75</td>\n",
       "      <td>40</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>3</td>\n",
       "      <td>446</td>\n",
       "      <td>322</td>\n",
       "      <td>2195</td>\n",
       "      <td>3</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>129</td>\n",
       "      <td>29</td>\n",
       "      <td>74</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>314</td>\n",
       "      <td>329</td>\n",
       "      <td>2150</td>\n",
       "      <td>2</td>\n",
       "      <td>293</td>\n",
       "      <td>317</td>\n",
       "      <td>2121</td>\n",
       "      <td>2</td>\n",
       "      <td>-3</td>\n",
       "      <td>36</td>\n",
       "      <td>29</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>230</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>367</td>\n",
       "      <td>327</td>\n",
       "      <td>2196</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>121</td>\n",
       "      <td>80</td>\n",
       "      <td>153</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>231</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>410</td>\n",
       "      <td>351</td>\n",
       "      <td>2347</td>\n",
       "      <td>3</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>164</td>\n",
       "      <td>104</td>\n",
       "      <td>304</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>232</th>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>2</td>\n",
       "      <td>430</td>\n",
       "      <td>365</td>\n",
       "      <td>2115</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>184</td>\n",
       "      <td>118</td>\n",
       "      <td>72</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>233</th>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>283</td>\n",
       "      <td>220</td>\n",
       "      <td>2129</td>\n",
       "      <td>1</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>37</td>\n",
       "      <td>-27</td>\n",
       "      <td>86</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>234</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>349</td>\n",
       "      <td>326</td>\n",
       "      <td>2081</td>\n",
       "      <td>2</td>\n",
       "      <td>247</td>\n",
       "      <td>246</td>\n",
       "      <td>2043</td>\n",
       "      <td>1</td>\n",
       "      <td>103</td>\n",
       "      <td>79</td>\n",
       "      <td>38</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>235 rows × 15 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     HOME_TEAM_SCORE  AWAY_TEAM_SCORE  result  home_attack  home_defense  \\\n",
       "0                  6                3       3          529           284   \n",
       "1                  3                2       3          392           333   \n",
       "2                  1                0       3          392           333   \n",
       "3                  2                0       3          446           322   \n",
       "4                  3                1       3          314           329   \n",
       "..               ...              ...     ...          ...           ...   \n",
       "230                1                1       2          367           327   \n",
       "231                1                1       2          410           351   \n",
       "232                4                4       2          430           365   \n",
       "233                1                2       1          283           220   \n",
       "234                0                1       1          349           326   \n",
       "\n",
       "     home_elo_rating  home_cluster  away_attack  away_defense  \\\n",
       "0               2352             3          293           317   \n",
       "1               2150             2          293           317   \n",
       "2               2150             2          293           317   \n",
       "3               2195             3          293           317   \n",
       "4               2150             2          293           317   \n",
       "..               ...           ...          ...           ...   \n",
       "230             2196             2          247           246   \n",
       "231             2347             3          247           246   \n",
       "232             2115             2          247           246   \n",
       "233             2129             1          247           246   \n",
       "234             2081             2          247           246   \n",
       "\n",
       "     away_elo_rating  away_cluster  attack_effective  defense_effective  \\\n",
       "0               2121             2               212                 -9   \n",
       "1               2121             2                75                 40   \n",
       "2               2121             2                75                 40   \n",
       "3               2121             2               129                 29   \n",
       "4               2121             2                -3                 36   \n",
       "..               ...           ...               ...                ...   \n",
       "230             2043             1               121                 80   \n",
       "231             2043             1               164                104   \n",
       "232             2043             1               184                118   \n",
       "233             2043             1                37                -27   \n",
       "234             2043             1               103                 79   \n",
       "\n",
       "     elo_rating_diff  team_strength_diff  \n",
       "0                231                   1  \n",
       "1                 29                   0  \n",
       "2                 29                   0  \n",
       "3                 74                   1  \n",
       "4                 29                   0  \n",
       "..               ...                 ...  \n",
       "230              153                   1  \n",
       "231              304                   2  \n",
       "232               72                   1  \n",
       "233               86                   0  \n",
       "234               38                   1  \n",
       "\n",
       "[235 rows x 15 columns]"
      ]
     },
     "execution_count": 67,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def get_data(HOME_TEAM: str, AWAY_TEAM: str):\n",
    "    # Check that the teams are valid\n",
    "    valid_teams = set(team_data['Teams'])\n",
    "    if HOME_TEAM not in valid_teams or AWAY_TEAM not in valid_teams:\n",
    "        raise ValueError(\"Invalid team name\")\n",
    "\n",
    "    # Get the relevant data from team_data\n",
    "    home_stat = team_data.loc[team_data['Teams'] == HOME_TEAM]\n",
    "    away_stat = team_data.loc[team_data['Teams'] == AWAY_TEAM]\n",
    "\n",
    "    # Calculate the features\n",
    "    attack_effective = home_stat['Attacks'].values[0] - away_stat['Defense'].values[0]\n",
    "    defense_effective = home_stat['Defense'].values[0] - away_stat['Attacks'].values[0]\n",
    "    elo_rating_diff = home_stat['elo rating'].values[0] - away_stat['elo rating'].values[0]\n",
    "\n",
    "    # Construct the output DataFrame\n",
    "    data = pd.DataFrame([[attack_effective, defense_effective, elo_rating_diff]],\n",
    "                        columns=['attack_effective', 'defense_effective', 'elo_rating_diff'])\n",
    "    return data\n",
    "data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 81,
   "id": "2d8fc824",
   "metadata": {},
   "outputs": [],
   "source": [
    "import random\n",
    "\n",
    "def predict_winner(HOME_TEAM, AWAY_TEAM):\n",
    "    match_up = get_data(HOME_TEAM, AWAY_TEAM)\n",
    "    reverse_match_up = get_data(HOME_TEAM, AWAY_TEAM)\n",
    "   \n",
    "    match_result_clf = get_clf_model_output(match_up)\n",
    "    reverse_match_result_clf = 2-get_clf_model_output(reverse_match_up)\n",
    "    score = match_result_clf + reverse_match_result_clf\n",
    "    winner = HOME_TEAM\n",
    "    if score>2:\n",
    "        winner = HOME_TEAM\n",
    "    elif score<2:\n",
    "        winner = AWAY_TEAM\n",
    "    else:\n",
    "        winner = random.choice([HOME_TEAM, AWAY_TEAM])\n",
    "    return winner"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 82,
   "id": "60a81874",
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_data(HOME_TEAM, AWAY_TEAM):\n",
    "    home_attack = team_data[team_data['Teams']==HOME_TEAM]['Attacks'].values[0]\n",
    "    home_defense = team_data[team_data['Teams']==HOME_TEAM]['Defense'].values[0]\n",
    "    home_elo_rating = team_data[team_data['Teams']==HOME_TEAM]['elo rating'].values[0]\n",
    "    home_cluster = team_data[team_data['Teams']==HOME_TEAM]['Cluster'].values[0]\n",
    "    away_attack = team_data[team_data['Teams']==AWAY_TEAM]['Attacks'].values[0]\n",
    "    away_defense = team_data[team_data['Teams']==AWAY_TEAM]['Defense'].values[0]\n",
    "    away_elo_rating = team_data[team_data['Teams']==AWAY_TEAM]['elo rating'].values[0]\n",
    "    away_cluster = team_data[team_data['Teams']==AWAY_TEAM]['Cluster'].values[0]\n",
    "    attack_effective = home_attack - away_defense\n",
    "    defense_effective = home_defense - away_attack\n",
    "    elo_rating_diff = home_elo_rating - away_elo_rating\n",
    "\n",
    "    data = pd.DataFrame([[attack_effective, defense_effective, elo_rating_diff]],\n",
    "                      columns=['attack_effective', 'defense_effective', 'elo_rating_diff'])\n",
    "    return data\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 83,
   "id": "a9c60105",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "import random\n",
    "\n",
    "def predict_winner(HOME_TEAM, AWAY_TEAM):\n",
    "    match_up = get_data(HOME_TEAM, AWAY_TEAM)\n",
    "    reverse_match_up = get_data(AWAY_TEAM, HOME_TEAM)\n",
    "   \n",
    "    match_result_clf = get_clf_model_output(match_up)\n",
    "    reverse_match_result_clf = 2-get_clf_model_output(reverse_match_up)\n",
    "    score = match_result_clf + reverse_match_result_clf\n",
    "    \n",
    "    winner = HOME_TEAM if score > 2 else (AWAY_TEAM if score < 2 else random.choice([HOME_TEAM, AWAY_TEAM]))\n",
    "    return winner\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 84,
   "id": "69de553e",
   "metadata": {},
   "outputs": [],
   "source": [
    "X = data[['attack_effective', 'defense_effective', 'elo_rating_diff']]\n",
    "y = data[['HOME_TEAM_SCORE', 'AWAY_TEAM_SCORE', 'result']]\n",
    "\n",
    "def get_clf_model_output(match_up):\n",
    "    y_pred = clf.predict(match_up)\n",
    "    return y_pred[:,2][0]\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 87,
   "id": "dfdd47f2",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Inter\n",
      "Bayern\n",
      "Real Madrid\n",
      "Napoli\n"
     ]
    }
   ],
   "source": [
    "qf1 = predict_winner('Benfica','Inter')\n",
    "print(qf1)\n",
    "qf2 = predict_winner('Man City','Bayern')\n",
    "print(qf2)\n",
    "qf3 = predict_winner('Real Madrid','Chelsea')\n",
    "print(qf3)\n",
    "qf4 = predict_winner('Napoli','Milan')\n",
    "print(qf4)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}