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)