即日起在codingBlog上分享您的技术经验即可获得积分,积分可兑换现金哦。

sklearn – tree

编程语言 u011906308 12℃ 0评论
class BaseDecisionTree
    """
    决策树的基础类
    """
    @abstractmethod
    def __init__
    def fit     
    def _validate_X_predict
    def predict
    def apply
    def decision_path

    @property
    def feature_importances_


class ClassifierMixin:
    """
    Mixin class for all classifiers in scikit-learn.
    """

    def score


class DecisionTreeClassifier(BaseDecisionTree, ClassifierMixin):
    """
    A decision tree classifier.
    """

    def __init__
    def fit:
        super(DecisionTreeClassifier, self).fit(X, y)

    def predict_proba
    def predict_log_proba





  • 用于拟合决策树模型的函数
BaseDecisionTree.fity = np.reshape(y, (-1, 1)) # 行转为列,且自动变为ndarray

    self.n_features_ = X.shape(1) # 特征个数
    self.n_outputs_ = y.shape[1] # 输出维度

    # 确认y是non-regression类型
    # sklearn.utils.multiclass
    check_classification_targets(y) 

    # 一个list
    # 里面的元素为ndarray,对应各个维度下类的种类,已排序
    self.classes_ 

    # 将其他水平转化为数字水平,从0开始
    y = np.unique(y, return_inverse=True)

转载请注明:CodingBlog » sklearn – tree

喜欢 (0)or分享 (0)
发表我的评论
取消评论

*

表情