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:19 +02:00
parent 3ae029a0e6
commit 58422eee10
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -57,7 +57,7 @@ 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)
loss, accuracy = model.evaluate(X_test, y_test) accuracy = model.evaluate(X_test, y_test)
print('Test accuracy:', accuracy) print('Test accuracy:', accuracy)
model.save(model_file) model.save(model_file)