Browse code

add 'data' to the SKOptLearner

Bas Nijholt authored on 16/10/2019 22:25:36
Showing 1 changed files
... ...
@@ -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):