Update 'simple_regression.py'

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
 )