1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/* Copyright 2015 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_CORE_LIB_HISTOGRAM_HISTOGRAM_H_
#define TENSORFLOW_CORE_LIB_HISTOGRAM_HISTOGRAM_H_
#include <string>
#include <vector>
#include "tensorflow/core/lib/gtl/array_slice.h"
#include "tensorflow/core/platform/macros.h"
#include "tensorflow/core/platform/mutex.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tensorflow/core/platform/types.h"
namespace tensorflow {
class HistogramProto;
namespace histogram {
class Histogram {
public:
// Create a histogram with a default set of bucket boundaries.
// Buckets near zero cover very small ranges (e.g. 10^-12), and each
// bucket range grows by ~10% as we head away from zero. The
// buckets cover the range from -DBL_MAX to DBL_MAX.
Histogram();
// Create a histogram with a custom set of bucket boundaries,
// specified in "custom_bucket_limits[0..custom_bucket_limits.size()-1]"
// REQUIRES: custom_bucket_limits[i] values are monotonically increasing.
// REQUIRES: custom_bucket_limits is not empty()
explicit Histogram(gtl::ArraySlice<double> custom_bucket_limits);
// Restore the state of a histogram that was previously encoded
// via Histogram::EncodeToProto. Note that only the bucket boundaries
// generated by EncodeToProto will be restored.
bool DecodeFromProto(const HistogramProto& proto);
~Histogram() {}
void Clear();
void Add(double value);
// Save the current state of the histogram to "*proto". If
// "preserve_zero_buckets" is false, only non-zero bucket values and
// ranges are saved, and the bucket boundaries of zero-valued buckets
// are lost.
void EncodeToProto(HistogramProto* proto, bool preserve_zero_buckets) const;
// Return the median of the values in the histogram
double Median() const;
// Return the "p"th percentile [0.0..100.0] of the values in the
// distribution
double Percentile(double p) const;
// Return the average value of the distribution
double Average() const;
// Return the standard deviation of values in the distribution
double StandardDeviation() const;
// Returns a multi-line human-readable string representing the histogram
// contents. Example output:
// Count: 4 Average: 251.7475 StdDev: 432.02
// Min: -3.0000 Median: 5.0000 Max: 1000.0000
// ------------------------------------------------------
// [ -5, 0 ) 1 25.000% 25.000% #####
// [ 0, 5 ) 1 25.000% 50.000% #####
// [ 5, 10 ) 1 25.000% 75.000% #####
// [ 1000, 10000 ) 1 25.000% 100.000% #####
std::string ToString() const;
private:
double min_;
double max_;
double num_;
double sum_;
double sum_squares_;
std::vector<double> custom_bucket_limits_;
gtl::ArraySlice<double> bucket_limits_;
std::vector<double> buckets_;
double Remap(double x, double x0, double x1, double y0, double y1) const;
TF_DISALLOW_COPY_AND_ASSIGN(Histogram);
};
// Wrapper around a Histogram object that is thread safe.
//
// All methods hold a lock while delegating to a Histogram object owned by the
// ThreadSafeHistogram instance.
//
// See Histogram for documentation of the methods.
class ThreadSafeHistogram {
public:
ThreadSafeHistogram() {}
explicit ThreadSafeHistogram(gtl::ArraySlice<double> custom_bucket_limits)
: histogram_(custom_bucket_limits) {}
bool DecodeFromProto(const HistogramProto& proto);
~ThreadSafeHistogram() {}
void Clear();
// TODO(touts): It might be a good idea to provide a AddN(<many values>)
// method to avoid grabbing/releasing the lock when adding many values.
void Add(double value);
void EncodeToProto(HistogramProto* proto, bool preserve_zero_buckets) const;
double Median() const;
double Percentile(double p) const;
double Average() const;
double StandardDeviation() const;
std::string ToString() const;
private:
mutable mutex mu_;
Histogram histogram_ GUARDED_BY(mu_);
};
} // namespace histogram
} // namespace tensorflow
#endif // TENSORFLOW_CORE_LIB_HISTOGRAM_HISTOGRAM_H_
|
