hugobuddel / orange3

Orange/classification/tree.py

import sklearn.tree as skl_tree

The import sklearn.tree could not be resolved.

from Orange.classification import SklLearner, SklModel
from Orange.preprocess import Continuize, RemoveNaNColumns, SklImpute

__all__ = ["TreeLearner"]


class TreeClassifier(SklModel):
    pass


class TreeLearner(SklLearner):
    __wraps__ = skl_tree.DecisionTreeClassifier
    __returns__ = TreeClassifier
    name = 'tree'
    preprocessors = [RemoveNaNColumns(),
                     SklImpute(),
                     Continuize()]

    def __init__(self, criterion="gini", splitter="best", max_depth=None,

The argument splitter seems to be unused.

The argument criterion seems to be unused.

The argument max_depth seems to be unused.

                 min_samples_split=2, min_samples_leaf=1,

The argument min_samples_split seems to be unused.

The argument min_samples_leaf seems to be unused.

                 max_features=None,

The argument max_features seems to be unused.

                 random_state=None, max_leaf_nodes=None,

The argument random_state seems to be unused.

The argument max_leaf_nodes seems to be unused.

                 preprocessors=None):
        super().__init__(preprocessors=preprocessors)
        self.params = vars()
        self.supports_weights = True