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):