Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils_test.py137 def custom_loss(y_obs, y_pred): function
143 model.compile('SGD', loss=custom_loss)
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format_test.py471 def custom_loss(y_true, y_pred): function
477 model.compile(loss=custom_loss, optimizer=CustomOp(), metrics=['acc'])
490 'custom_loss': custom_loss})