[Book Question] Machine Learning in Elixir: Poor accuracy for Chapter 1's example

You can see what the model is predicting for the test data with Axon.predict/4

Axon.predict(model, trained_model_state, x_test)