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

Final commit of the exercise 6

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-15 17:24:05 +02:00
parent 9cc3be30c1
commit bdfdd2f40f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Zadanie_06_training.py
View File

@ -4,6 +4,7 @@ from sklearn.metrics import accuracy_score, classification_report
import pandas as pd
from sklearn.model_selection import train_test_split
import numpy as np
import sys
from sklearn.preprocessing import StandardScaler, LabelEncoder
from tensorflow.keras.optimizers import Adam
@ -45,6 +46,6 @@ rms = Adam(lr=0.0003)
NeuralModel.compile(optimizer=rms, loss='binary_crossentropy', metrics=['accuracy'])
NeuralModel.fit(scaled_x, yenc, epochs=15) #verbose = 1
NeuralModel.fit(scaled_x, yenc, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
NeuralModel.save('wine_model.h5')