s464961/ium_464914
1
0
Fork 0
forked from s464914/ium_464914
Code Pull Requests Activity

Update model.py

Browse Source
This commit is contained in:
Alicja Szulecka 2024-04-29 21:47:03 +02:00
parent b45d036d42
commit 6a0b357945
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@ -31,7 +31,7 @@ class Model(nn.Module):
return x return x
def main(): def main():
epochs = sys.argv[1] epochs = int(sys.argv[1])
print(epochs) print(epochs)
forest_train = pd.read_csv('forest_train.csv') forest_train = pd.read_csv('forest_train.csv')