add_custom_metric
The purpose of the
leap_binder.add_custom_metric function is to add a custom metric suited to your needs to the tensorleap platform.code_loader.leap_binder.add_custom_metric(
function=Callable[..., tf.Tensor],
name=str
)
Args | Text |
|---|---|
function | A function that returns the required metrics |
name | The name of the the metric |
import tensorflow as tf
def custom_metric_mean_distance(y_true, y_pred):
rank = len(y_true.shape)
axis = range(1, rank)
diff = y_true - y_pred
return tf.reduce_mean(diff, axis=axis)
leap_binder.add_custom_metric(custom_metric_mean_distance, "mean distance")
Last modified 2mo ago