s434788/ium_434788
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

next htofix

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-15 15:08:39 +02:00
parent c94d5f815e
commit 095d8aead2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Zadanie_06_training.py
View File

@ -31,6 +31,6 @@ def regression_model():
return model
model = regression_model()
model.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = sys.argv[2]) #verbose = 1
model.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
model.save('wine_model')