s434700/ium_s434700
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

machine learning

Browse Source
This commit is contained in:
Filip Izydorczyk 2021-04-26 06:17:28 +02:00
parent 5aec2ffb4e
commit 30396e4036
2 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile
View File

@ -6,5 +6,10 @@ RUN pacman -S python-pip --noconfirm
RUN pip install --user kaggle RUN pip install --user kaggle
RUN pip install --user pandas 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 COPY ./data.sh /home/data/data.sh
COPY ./ml.py /home/data/ml.py

19
ml.py Normal file
View File

@ -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)