Ask or search…
K

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

Examples

Basic Usage

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")
Full script usage can be found at Integration Script.