示例#1
0
文件: mean.py 项目: Jasputtar/deepr
 def __call__(self, tensors: Dict[str, tf.Tensor]) -> Dict[str, Tuple]:
     tensors = base.get_scalars(tensors,
                                names=self.tensors,
                                pattern=self.pattern)
     return {
         name: decay_mean_metric(value, self.decay, name)
         for name, value in tensors.items()
     }
示例#2
0
文件: mean.py 项目: Jasputtar/deepr
 def __call__(self, tensors: Dict[str, tf.Tensor]) -> Dict[str, Tuple]:
     tensors = base.get_scalars(tensors,
                                names=self.tensors,
                                pattern=self.pattern)
     return {
         name: tf.compat.v1.metrics.mean(value)
         for name, value in tensors.items()
     }
示例#3
0
 def __call__(self, tensors: Dict[str, tf.Tensor]) -> Dict[str, Tuple]:
     tensors = base.get_scalars(tensors,
                                names=self.tensors,
                                pattern=self.pattern)
     LOGGER.info(f"{self} -> {', '.join(tensors.keys())}")
     return {
         name: max_value_metric(value, name)
         for name, value in tensors.items()
     }
示例#4
0
 def __call__(self, tensors: Dict[str, tf.Tensor]) -> Dict[str, Tuple]:
     tensors = base.get_scalars(tensors, names=self.tensors, pattern=self.pattern)
     LOGGER.info(f"{self} -> {', '.join(tensors.keys())}")
     return {name: (tensor, tf.no_op()) for name, tensor in tensors.items()}