Update 'simple_regression_lab7.py'
All checks were successful
s449288-evaluation/pipeline/head This commit looks good
s449288-training/pipeline/head This commit looks good

This commit is contained in:
Kacper Dudzic 2022-05-05 23:04:53 +02:00
parent a034fa5664
commit c48ab0ae5c

View File

@ -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')