Keras: .predict returns percentages instead of classes

by

You did not do anything wrong, predict has always returned the output of the model, for a classifier this has always been probabilities per class.

predict_classes is only available for Sequential models, not for Functional ones.

But there is an easy solution, you just need to take the argmax on the last dimension and you will get class indices:

y_probs = model1.predict(test_text_matrix)
y_pred  = np.argmax(y_probs, axis=-1)

More Related Contents:

  1. Keras, How to get the output of each layer?
  2. Keras Dense layer’s input is not flattened
  3. Tensorflow – ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type float)
  4. Deep-Learning Nan loss reasons
  5. Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,
  6. Why is TensorFlow 2 much slower than TensorFlow 1?
  7. Can Keras with Tensorflow backend be forced to use CPU or GPU at will?
  8. Proper way to feed time-series data to stateful LSTM?
  9. What is the difference between tf.keras and tf.python.keras?
  10. How to use a Keras RNN model to forecast for future dates or events?
  11. TypeError: __array__() takes 1 positional argument but 2 were given (Image classification Keras)
  12. Tensorflow 2.0 – AttributeError: module ‘tensorflow’ has no attribute ‘Session’
  13. Can I run Keras model on gpu?
  14. Keras custom loss function: Accessing current input pattern
  15. Negative dimension size caused by subtracting 3 from 1 for ‘conv2d_2/convolution’
  16. Dimension of shape in conv1D
  17. How to know if underfitting or overfitting is occuring?
  18. Loading a trained Keras model and continue training
  19. Keras Custom loss function to pass arguments other than y_true and y_pred
  20. Unknown initializer: GlorotUniform when loading Keras model
  21. Tensorflow : logits and labels must have the same first dimension
  22. Multiple outputs in Keras
  23. How to implement Grad-CAM on a trained network
  24. How to fix “AttributeError: module ‘tensorflow’ has no attribute ‘get_default_graph'”?
  25. ‘Tensor’ object has no attribute ‘lower’
  26. Unexpected keyword argument ‘ragged’ in Keras
  27. ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`
  28. How do I plot a Keras/Tensorflow subclassing API model?
  29. Strange behaviour of the loss function in keras model, with pretrained convolutional base
  30. Invalid Argument Error / Graph Execution Error

Leave a Comment