Решение пень в Python - PullRequest
       35

Решение пень в Python

0 голосов
/ 15 октября 2018

Я пытаюсь перенести код 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 должны быть включены категориальные предикторы, но я не знаю как!)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...