add EPOCHS variable

This commit is contained in:
Sebastian Wałęsa 2022-05-06 20:17:41 +02:00
parent 8db0b2eae4
commit 7dd45b9c75

View File

@ -22,7 +22,7 @@ import sys
# In[2]: # In[2]:
EPOCHS = int(sys.argv[1])
#load data #load data
dataframe = pd.read_csv("understat.csv") dataframe = pd.read_csv("understat.csv")
@ -124,7 +124,7 @@ model=Model_xPosition()
# In[11]: # In[11]:
epochs = 1000 epochs = EPOCHS
lr = 1e-5 lr = 1e-5
learning_proccess = fit(epochs, lr, model, train_loader, val_loader) learning_proccess = fit(epochs, lr, model, train_loader, val_loader)