Навчіть і оцініть модель

Просунуте глибоке навчання з Keras

Zach Deane-Mayer

Data Scientist

Дані з баскетболу

Мета: передбачити результати турніру

Доступні дані: рейтинги команд від організаторів турніру

import pandas as pd
games_tourney = pd.read_csv('datasets/games_tourney.csv')
games_tourney.head()

Out[1]:
   season  team_1  team_2  home  seed_diff  score_diff  score_1  score_2  won
0    1985     288      73     0         -3          -9       41       50    0
1    1985    5929      73     0          4           6       61       55    1
2    1985    9884      73     0          5          -4       59       63    0
3    1985      73     288     0          3           9       50       41    1
4    1985    3920     410     0          1          -9       54       63    0

Дані з баскетболу

Вхід: різниця посівів

import pandas as pd
games_tourney = pd.read_csv('datasets/games_tourney.csv')
games_tourney.head()

Дані з баскетболу

Вихід: різниця в рахунку

import pandas as pd
games_tourney = pd.read_csv('datasets/games_tourney.csv')
games_tourney.head()

Дані з баскетболу

Вхід:

Різниця посівів — одне число: від -15 до +15
Посіви від 1 до 16
Найбільша різниця: 16–1 = +15
Найменша різниця: 1–16 = -15

Вихід:

Різниця в рахунку — одне число: від -50 до +50

Дані з баскетболу

Різниця посівів: 15
- Команда 1: 16
- Команда 2: 1
Різниця посівів: -15
- Команда 1: 1
- Команда 2: 16

Дані з баскетболу

Різниця в рахунку: -9
- Команда 1: 41
- Команда 2: 50
Різниця в рахунку: 6
- Команда 1: 61
- Команда 2: 55

Дані з баскетболу

import pandas as pd
games_tourney = pd.read_csv('datasets/games_tourney_samp.csv')
games_tourney.head()

Out[1]:
   season  team_1  team_2  home  seed_diff  score_diff  score_1  score_2  won
0    2017     320    6323     0         13          18      100       82    1
1    2017    6323     320     0        -13         -18       82      100    0

Побудуйте модель

from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, Dense
input_tensor = Input(shape=(1,))
output_tensor = Dense(1)(input_tensor)
model = Model(input_tensor, output_tensor)
model.compile(optimizer='adam', loss='mae')

Навчіть модель

from pandas import read_csv
games = read_csv('datasets/games_tourney.csv')
model.fit(games['seed_diff'],
          games['score_diff'],
          batch_size=64,
          validation_split=.20,
          verbose=True)

Оцініть модель

model.evaluate(games_test['seed_diff'],
               games_test['score_diff'])

1000/1000 [==============================] - 0s 26us/step
Out[1]: 9.145421981811523

Давайте потренуємось!

Просунуте глибоке навчання з Keras