[tensorflow] How to use contrib.eager.metrics on tensorflow 2.x version

 

 

 

# version 1.x
from tensorflow import contrib
tfe = contrib.eager

epoch_loss_avg = tfe.metrics.Mean()
epoch_accuracy = tfe.metrics.Accuracy()

 

to

 

# version 2.x
epoch_loss_avg = tf.keras.metrics.Mean()
epoch_accuracy = tf.keras.metrics.Accuracy()