From c48ab0ae5c6db369363a648faa37077117e700b7 Mon Sep 17 00:00:00 2001
From: Kacper Dudzic <kacdud1@st.amu.edu.pl>
Date: Thu, 5 May 2022 23:04:53 +0200
Subject: [PATCH] Update 'simple_regression_lab7.py'

---
 simple_regression_lab7.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/simple_regression_lab7.py b/simple_regression_lab7.py
index eac89bb..b01a8be 100644
--- a/simple_regression_lab7.py
+++ b/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')