Update 'simple_regression_lab7.py'
This commit is contained in:
parent
a034fa5664
commit
c48ab0ae5c
@ -20,14 +20,14 @@ ex.observers.append(FileStorageObserver('runs'))
|
|||||||
# Przykładowa modyfikowalna z Sacred konfiguracja wybranych parametrów treningu
|
# Przykładowa modyfikowalna z Sacred konfiguracja wybranych parametrów treningu
|
||||||
@ex.config
|
@ex.config
|
||||||
def config():
|
def config():
|
||||||
|
epochs = 100
|
||||||
units = 1
|
units = 1
|
||||||
learning_rate = 0.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
|
# Reszta kodu wrzucona do udekorowanej funkcji train do wywołania przez Sacred, żeby coś było capture'owane
|
||||||
@ex.capture
|
@ex.capture
|
||||||
def train(units, learning_rate, _run):
|
def train(epochs, units, learning_rate, _run):
|
||||||
|
|
||||||
# Wczytanie danych
|
# Wczytanie danych
|
||||||
data_train = pd.read_csv('lego_sets_clean_train.csv')
|
data_train = pd.read_csv('lego_sets_clean_train.csv')
|
||||||
|
Loading…
Reference in New Issue
Block a user