class Tensorflow::Keras::Metrics::SparseCategoricalAccuracy
Public Instance Methods
update_state(y_true, y_pred)
click to toggle source
Calls superclass method
Tensorflow::Keras::Metrics::Mean#update_state
# 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