Python中的决策树桩

时间:2018-10-15 19:44:47

标签: python matlab scikit-learn decision-tree

我正在尝试将Matlab代码传输到Python,并获得完全相同的结果。在Matlab代码中,我有一个决策树桩,如下所示:

switch Model
    case 'DecisionStump'

        Weights = Pt; % Train the weak learner by weights Pt

        tree = fitctree(X,Y,'minparent',size(X,1)-sum(Weights==0),'prune','off','mergeleaves','off','Weights', Weights, 'CategoricalPredictors', CategoricalPredictors);

        h = compact(tree);

并且我使用下面的python代码构造完全相同的决策树桩:

clf_tree = DecisionTreeClassifier(max_depth = 1)

但是,通过这两个程序,我得到的结果略有不同,如果我错过任何事情,如果有人可以帮助我,那将是非常不错的。 (也许应该在我的python代码中包含分类预测变量,但我不知道怎么做!)

0 个答案:

没有答案