class Tensorflow::Keras::Metrics::SparseCategoricalAccuracy

Public Instance Methods

update_state(y_true, y_pred) click to toggle source
# File lib/tensorflow/keras/metrics/sparse_categorical_accuracy.rb, line 5
def update_state(y_true, y_pred)
  y_pred = RawOps.arg_max(y_pred, -1)

  # if y_pred.dtype != y_true.dtype
  #   y_pred = Tensorflow.cast(y_pred, y_true.dtype)
  # end

  super(Math.equal(y_true, y_pred))
end