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

Final commit of the exercise 6_v2

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-15 17:28:18 +02:00
parent bdfdd2f40f
commit fd95b53f46
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

@ -46,6 +46,6 @@ rms = Adam(lr=0.0003)
NeuralModel.compile(optimizer=rms, loss='binary_crossentropy', metrics=['accuracy'])
NeuralModel.fit(scaled_x, yenc, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
NeuralModel.fit(scaled_x, yenc, batch_size=int(sys.argv[1]), epochs = int(sys.argv[2])) #verbose = 1
NeuralModel.save('wine_model.h5')