|
2 | 2 | # pylint: disable=invalid-sequence-index |
3 | 3 |
|
4 | 4 | import sys |
5 | | -import functools |
6 | 5 | from itertools import chain |
7 | 6 | import abc |
8 | 7 | import enum |
|
13 | 12 |
|
14 | 13 | import concurrent.futures |
15 | 14 | from concurrent.futures import Future |
16 | | - |
17 | 15 | from collections import OrderedDict, namedtuple |
18 | 16 |
|
19 | 17 | try: |
|
31 | 29 | from AnyQt.QtCore import Qt, QSize, QThread, QMetaObject, Q_ARG |
32 | 30 | from AnyQt.QtCore import pyqtSlot as Slot |
33 | 31 |
|
| 32 | +from Orange.base import Learner |
| 33 | +import Orange.classification |
34 | 34 | from Orange.data import Table, DiscreteVariable, ContinuousVariable |
| 35 | +from Orange.data.filter import HasClass |
35 | 36 | from Orange.data.sql.table import SqlTable, AUTO_DL_LIMIT |
36 | 37 | import Orange.evaluation |
37 | | -import Orange.classification |
38 | | -import Orange.regression |
39 | | - |
40 | | -from Orange.base import Learner |
41 | 38 | from Orange.evaluation import scoring, Results |
42 | 39 | from Orange.preprocess.preprocess import Preprocess |
43 | | -from Orange.preprocess import RemoveNaNClasses |
| 40 | +import Orange.regression |
44 | 41 | from Orange.widgets import gui, settings, widget |
45 | 42 | from Orange.widgets.utils.itemmodels import DomainModel |
46 | 43 | from Orange.widgets.widget import OWWidget, Msg, Input, Output |
47 | 44 | from Orange.widgets.utils.concurrent import ThreadExecutor |
48 | 45 |
|
| 46 | + |
49 | 47 | log = logging.getLogger(__name__) |
50 | 48 |
|
51 | 49 | InputLearner = namedtuple( |
@@ -389,7 +387,7 @@ def set_train_data(self, data): |
389 | 387 | if self.train_data_missing_vals or self.test_data_missing_vals: |
390 | 388 | self.Warning.missing_data(self._which_missing_data()) |
391 | 389 | if data: |
392 | | - data = RemoveNaNClasses(data) |
| 390 | + data = HasClass(data) |
393 | 391 | else: |
394 | 392 | self.Warning.missing_data.clear() |
395 | 393 |
|
@@ -439,7 +437,7 @@ def set_test_data(self, data): |
439 | 437 | if self.train_data_missing_vals or self.test_data_missing_vals: |
440 | 438 | self.Warning.missing_data(self._which_missing_data()) |
441 | 439 | if data: |
442 | | - data = RemoveNaNClasses()(data) |
| 440 | + data = HasClass()(data) |
443 | 441 | else: |
444 | 442 | self.Warning.missing_data.clear() |
445 | 443 |
|
|
0 commit comments