Update 'simple_regression_lab7.py'

simple_regression_lab7.py

@@ -20,14 +20,14 @@ ex.observers.append(FileStorageObserver('runs'))
 # Przykładowa modyfikowalna z Sacred konfiguracja wybranych parametrów treningu
 @ex.config
 def config():
+    epochs = 100
     units = 1
     learning_rate = 0.1
-    epochs = 100
 
 
 # Reszta kodu wrzucona do udekorowanej funkcji train do wywołania przez Sacred, żeby coś było capture'owane
 @ex.capture
-def train(units, learning_rate, _run):
+def train(epochs, units, learning_rate, _run):
 
     # Wczytanie danych
     data_train = pd.read_csv('lego_sets_clean_train.csv')