From 7445be701468b63a31b4c816f7e895e383e47392 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 8 May 2022 20:33:07 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytroch_sacred.py' --- ml_pytroch_sacred.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ml_pytroch_sacred.py b/ml_pytroch_sacred.py index 2373faa..f4be1cc 100644 --- a/ml_pytroch_sacred.py +++ b/ml_pytroch_sacred.py @@ -21,7 +21,7 @@ ex.observers.append(FileStorageObserver('IUM_478839')) @ex.config def my_config(): - epochs = 100 + epochs = 1000 #load data dataframe = pd.read_csv("understat.csv") @@ -104,7 +104,6 @@ def fit(epochs, lr, model, train_loader, val_loader, opt_func=torch.optim.SGD): input_size = len(input_cols) output_size = len(output_cols) model=Model_xPosition() -epochs = 1000 lr = 1e-5 learning_proccess = fit(epochs, lr, model, train_loader, val_loader) def predict_single(input, target, model):