def _get_measures(self, targets, predictions):

        with tf.variable_scope('measures_{}'.format(self.name)):
            error_val = get_error(targets, predictions, self.name)

            absolute_error_val = get_absolute_error(targets, predictions,
                                                    self.name)

            squared_error_val = get_squared_error(targets, predictions,
                                                  self.name)

            r2_val = get_r2(targets, predictions, self.name)

        return error_val, squared_error_val, absolute_error_val, r2_val
示例#2
0
    def _get_measures(self, targets, predictions):

        with tf.variable_scope('measures_{}'.format(self.name)):
            error_val = get_error(targets, predictions, self.name)

            absolute_error_val = tf.reduce_sum(get_absolute_error(
                targets, predictions, self.name),
                                               axis=1)

            squared_error_val = tf.reduce_sum(get_squared_error(
                targets, predictions, self.name),
                                              axis=1)

            # TODO - not sure if this is correct
            r2_val = tf.reduce_sum(get_r2(targets, predictions, self.name))

        return error_val, squared_error_val, absolute_error_val, r2_val