diff --git a/train.py b/train.py
index 64d80a9..72ff481 100644
--- a/train.py
+++ b/train.py
@@ -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.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 loss:', loss)
 
     model.save(model_file)