Searched refs:weight_key (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | metric_spec_test.py | 61 metric_fn=fn, prediction_key="p1", label_key="l1", weight_key="f2")
226 spec = MetricSpec(metric_fn=_fn, label_key="l1", weight_key="f2")
242 spec = MetricSpec(metric_fn=_fn, prediction_key="p1", weight_key="f2")
260 spec = MetricSpec(metric_fn=_fn, label_key="l1", weight_key="f2")
276 spec = MetricSpec(metric_fn=_fn, prediction_key="p1", weight_key="f2")
291 metric_fn=_fn, prediction_key="p1", label_key="l1", weight_key="f2")
307 metric_fn=_fn, prediction_key="p1", label_key="l1", weight_key="f2")
320 weight_key="my_weight"))
335 weight_key="my_weight"))
358 weight_key="f2")
[all …]
|
| D | metric_spec.py | 302 weight_key=None): argument
332 is_weights_required=weight_key is not None)
335 self._weight_key = weight_key
346 def weight_key(self): member in MetricSpec
366 'weight_key=%s)' % self.weight_key
439 weights=inputs[self.weight_key] if self.weight_key else None)
|