1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from Orange.classification import SklTreeLearner
from Orange.classification import TreeLearner as ClassificationTreeLearner
from Orange.modelling import Fitter, SklFitter
from Orange.regression import TreeLearner as RegressionTreeLearner
from Orange.regression.tree import SklTreeRegressionLearner
from Orange.tree import TreeModel
__all__ = ['SklTreeLearner', 'TreeLearner']
class SklTreeLearner(SklFitter):
name = 'tree'
__fits__ = {'classification': SklTreeLearner,
'regression': SklTreeRegressionLearner}
class TreeLearner(Fitter):
name = 'tree'
__fits__ = {'classification': ClassificationTreeLearner,
'regression': RegressionTreeLearner}
__returns__ = TreeModel
|
