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