... | ... |
@@ -26,10 +26,12 @@ class SKOptLearner(Optimizer, BaseLearner): |
26 | 26 |
def __init__(self, function, **kwargs): |
27 | 27 |
self.function = function |
28 | 28 |
self.pending_points = set() |
29 |
+ self.data = {} |
|
29 | 30 |
super().__init__(**kwargs) |
30 | 31 |
|
31 | 32 |
def tell(self, x, y, fit=True): |
32 | 33 |
self.pending_points.discard(x) |
34 |
+ self.data[x] = y |
|
33 | 35 |
super().tell([x], y, fit) |
34 | 36 |
|
35 | 37 |
def tell_pending(self, x): |