diff --git a/simple_regression.py b/simple_regression.py
index 6837942..b45d523 100644
--- a/simple_regression.py
+++ b/simple_regression.py
@@ -4,6 +4,10 @@ from keras.models import save_model
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
+import sys
+
+# Pobranie przykładowego argumentu trenowania
+EPOCHS_NUM = int(sys.argv[1])
 
 # Wczytanie danych
 data_train = pd.read_csv('lego_sets_clean_train.csv')
@@ -35,7 +39,7 @@ model.compile(
 history = model.fit(
     train_piece_counts,
     train_prices,
-    epochs=100,
+    epochs=EPOCHS_NUM,
     verbose=0,
     validation_split=0.2
 )