s444517/ium_444517
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

correcting training file
Some checks failed
s444517-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Kamila 2022-05-03 11:16:22 +02:00
parent 7075223969
commit 9e17635330
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nn_train.py
View File

@ -46,6 +46,11 @@ encoder.fit(y_test_set)
encoded_Ytt = encoder.transform(y_test_set)
dummy_ytt = np_utils.to_categorical(encoded_Ytt)
try:
no_epochs=int(sys.argv[1])
except:
no_epochs = 200
# model definition
number_of_classes = 33
number_of_features = 5
@ -53,7 +58,7 @@ model = Sequential()
model.add(Dense(number_of_classes, activation='relu'))
model.add(Dense(number_of_classes, activation='softmax',input_dim=number_of_features))
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy', 'categorical_accuracy'])
model.fit(x_train_set, dummy_y, epochs=int(sys.argv[1]), validation_data=(x_validate_set, dummy_yv))
model.fit(x_train_set, dummy_y, epochs=no_epochs, validation_data=(x_validate_set, dummy_yv))
model.save("my_model/")
#model predictions