Module imodels.tree.gosdt_wrapper

Expand source code
# This is just a simple wrapper around gosdt: https://github.com/Jimmy-Lin/GeneralizedOptimalSparseDecisionTrees

from sklearn.tree import DecisionTreeClassifier


class GlobalSparseTreeClassifier(DecisionTreeClassifier):
    """Placeholder for GOSDT classifier
    """

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def __str__(self):
        return 'Global Sparse Tree ' + str(self.tree_)

Classes

class GlobalSparseTreeClassifier (*args, **kwargs)

Placeholder for GOSDT classifier

Expand source code
class GlobalSparseTreeClassifier(DecisionTreeClassifier):
    """Placeholder for GOSDT classifier
    """

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def __str__(self):
        return 'Global Sparse Tree ' + str(self.tree_)

Ancestors

  • sklearn.tree._classes.DecisionTreeClassifier
  • sklearn.base.ClassifierMixin
  • sklearn.tree._classes.BaseDecisionTree
  • sklearn.base.MultiOutputMixin
  • sklearn.base.BaseEstimator