s478839/ium_478839
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add EPOCHS variable

Browse Source
This commit is contained in:
Sebastian Wałęsa 2022-05-06 20:17:41 +02:00
parent 8db0b2eae4
commit 7dd45b9c75
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ml_pytorch.py
View File

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