Browse code
fix _get_data for the BalancingLearner (#150)
Bas Nijholt authored on 29/01/2019 15:11:12
• GitHub committed on 29/01/2019 15:11:12
Showing 1 changed files
| ... | ... |
@@ -381,7 +381,7 @@ class BalancingLearner(BaseLearner): |
| 381 | 381 |
l.load(fname(l), compress=compress) |
| 382 | 382 |
|
| 383 | 383 |
def _get_data(self): |
| 384 |
- return [l._get_data() for l in learner.learners] |
|
| 384 |
+ return [l._get_data() for l in self.learners] |
|
| 385 | 385 |
|
| 386 | 386 |
def _set_data(self, data): |
| 387 | 387 |
for l, _data in zip(self.learners, data): |