... | ... |
@@ -440,3 +440,16 @@ class BalancingLearner(BaseLearner): |
440 | 440 |
def _set_data(self, data): |
441 | 441 |
for l, _data in zip(self.learners, data): |
442 | 442 |
l._set_data(_data) |
443 |
+ |
|
444 |
+ def __getstate__(self): |
|
445 |
+ return ( |
|
446 |
+ self.learners, |
|
447 |
+ self._cdims_default, |
|
448 |
+ self.strategy, |
|
449 |
+ self._get_data(), |
|
450 |
+ ) |
|
451 |
+ |
|
452 |
+ def __setstate__(self, state): |
|
453 |
+ learners, cdims, strategy, data = state |
|
454 |
+ self.__init__(learners, cdims=cdims, strategy=strategy) |
|
455 |
+ self._set_data(data) |