s487187/ium_487187
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s

Browse Source
This commit is contained in:
Jakub Zaręba 2023-05-10 22:03:45 +02:00
parent 58422eee10
commit ba6e518100
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -57,8 +57,9 @@ def train_model(data_file, model_file, epochs, batch_size, test_size, random_sta
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy']) model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size) model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size)
accuracy = model.evaluate(X_test, y_test) loss, accuracy = model.evaluate(X_test, y_test)
print('Test accuracy:', accuracy) print('Test accuracy:', accuracy)
print('Test loss:', loss)
model.save(model_file) model.save(model_file)