Home
last modified time | relevance | path

Searched refs:CustomMetric (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py1087 class CustomMetric(base_cls): class
1093 super(CustomMetric, self).update_state(*args)
1096 metric = CustomMetric()
1107 with generic_utils.CustomObjectScope({'CustomMetric': CustomMetric}):
1165 class CustomMetric(keras.metrics.MeanSquaredError): class
1169 model.compile(loss='mse', optimizer='rmsprop', metrics=[CustomMetric()])