Searched defs:metric_fn (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
H A Dextenders_test.py56 def metric_fn(features): function in function:AddMetricsTest.test_should_add_metrics
59 estimator = extenders.add_metrics(estimator, metric_fn)
71 def metric_fn(features, not_recognized): function in function:AddMetricsTest.test_should_error_out_for_not_recognized_args
76 estimator = extenders.add_metrics(estimator, metric_fn)
82 def metric_fn(features, predictions, labels, config): function in function:AddMetricsTest.test_all_supported_args
89 estimator = extenders.add_metrics(estimator, metric_fn)
98 def metric_fn(labels, config, features, predictions): function in function:AddMetricsTest.test_all_supported_args_in_different_order
105 estimator = extenders.add_metrics(estimator, metric_fn)
114 def metric_fn(): function in function:AddMetricsTest.test_all_args_are_optional
117 estimator = extenders.add_metrics(estimator, metric_fn)
130 def metric_fn(): function in function:AddMetricsTest.test_overrides_existing_metrics
[all...]
/external/tensorflow/tensorflow/contrib/learn/python/learn/
H A Dmetric_spec.py30 '`metric_fn` requires named args: '
107 metric_fn, metric_fn_name, is_labels_required, is_weights_required):
108 """Adapt `metric_fn` to take only named args.
111 and `weights`, and invokes `metric_fn` according to the following rules:
112 - If `metric_fn` args include exactly one of `_LABELS_ARGS`, that arg is
115 - If `metric_fn` args include exactly one of `_PREDICTIONS_ARGS`, that arg is
122 metric_fn: Metric function to be wrapped.
123 metric_fn_name: `metric_fn` name, only used for logging.
129 and passing those to `metric_fn`.
133 - `metric_fn` ha
339 def metric_fn(self): member in class:MetricSpec
[all...]

Completed in 395 milliseconds