From b061491605f86c5509565fb09330163f10d1e5f8 Mon Sep 17 00:00:00 2001
From: Kacper Dudzic <kacdud1@st.amu.edu.pl>
Date: Sun, 24 Apr 2022 18:05:57 +0200
Subject: [PATCH] Update 'simple_regression.py'

---
 simple_regression.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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
 )