From 30396e40364cd65a62bd9cf6c55ca6bb1b4ee538 Mon Sep 17 00:00:00 2001 From: Filip Izydorczyk Date: Mon, 26 Apr 2021 06:17:28 +0200 Subject: [PATCH] machine learning --- Dockerfile | 7 ++++++- ml.py | 19 +++++++++++++++++++ 2 files changed, 25 insertions(+), 1 deletion(-) create mode 100644 ml.py diff --git a/Dockerfile b/Dockerfile index 0c308d0..a9250e2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -6,5 +6,10 @@ RUN pacman -S python-pip --noconfirm RUN pip install --user kaggle RUN pip install --user pandas +RUN pip install --user torch +RUN pip install --user numpy +RUN pip install --user matplotlib +RUN pip install --user seaborn -COPY ./data.sh /home/data/data.sh \ No newline at end of file +COPY ./data.sh /home/data/data.sh +COPY ./ml.py /home/data/ml.py \ No newline at end of file diff --git a/ml.py b/ml.py new file mode 100644 index 0000000..890ac82 --- /dev/null +++ b/ml.py @@ -0,0 +1,19 @@ +import torch +import torch.nn as nn +import numpy as np +import pandas as pd +import matplotlib.pyplot as plt +import seaborn as sns + +dataset = pd.read_csv('./train_set.csv') +print(dataset.head()) + +print(dataset.map_winner) + +# teams1 = dataset['team_1'].cat.codes.values +# teams2 = dataset['team_2'].cat.codes.values + +print(dataset.dtypes) + +device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") +print(device)