-
Notifications
You must be signed in to change notification settings - Fork 669
/
uniform_best.go
68 lines (57 loc) · 1.49 KB
/
uniform_best.go
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
// Copyright (C) 2019-2023, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package sampler
import (
"math"
"time"
"github.com/ava-labs/avalanchego/utils/timer/mockable"
)
var _ Uniform = (*uniformBest)(nil)
// Sampling is performed by using another implementation of the Uniform
// interface.
//
// Initialization attempts to find the best sampling algorithm given the dataset
// by performing a benchmark of the provided implementations.
type uniformBest struct {
Uniform
samplers []Uniform
maxSampleSize int
benchmarkIterations int
clock mockable.Clock
}
// NewBestUniform returns a new sampler
func NewBestUniform(expectedSampleSize int) Uniform {
return &uniformBest{
samplers: []Uniform{
&uniformReplacer{},
&uniformResample{},
},
maxSampleSize: expectedSampleSize,
benchmarkIterations: 100,
}
}
func (s *uniformBest) Initialize(length uint64) {
s.Uniform = nil
bestDuration := time.Duration(math.MaxInt64)
sampleSize := s.maxSampleSize
if length < uint64(sampleSize) {
sampleSize = int(length)
}
samplerLoop:
for _, sampler := range s.samplers {
sampler.Initialize(length)
start := s.clock.Time()
for i := 0; i < s.benchmarkIterations; i++ {
if _, err := sampler.Sample(sampleSize); err != nil {
continue samplerLoop
}
}
end := s.clock.Time()
duration := end.Sub(start)
if duration < bestDuration {
bestDuration = duration
s.Uniform = sampler
}
}
s.Uniform.Reset()
}