What is 'categorical predictor' in decision tree for regression

Question

Salad Box 2019-1-10

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/439225-what-is-categorical-predictor-in-decision-tree-for-regression

评论： Rui An 2020-11-23

Hi, I'd like to use Matlab's own example for the question. Please refer to https://uk.mathworks.com/help/stats/fitrtree.html for the original example.

>> load carsmall
>> whos
  Name                Size            Bytes  Class     Attributes
  Acceleration      100x1               800  double              
  Cylinders         100x1               800  double              
  Displacement      100x1               800  double              
  Horsepower        100x1               800  double              
  MPG               100x1               800  double              
  Mfg               100x13             2600  char                
  Model             100x33             6600  char                
  Model_Year        100x1               800  double              
  Origin            100x7              1400  char                
  Weight            100x1               800  double              
>> tree = fitrtree([Weight, Cylinders],MPG,...
                'categoricalpredictors',2,'MinParentSize',20,...
                'PredictorNames',{'W','C'})
tree = 
  RegressionTree
           PredictorNames: {'W'  'C'}
             ResponseName: 'Y'
    CategoricalPredictors: 2
        ResponseTransform: 'none'
          NumObservations: 94
  Properties, Methods

What exactly is the Categorical Predictors in this case and why it is 2?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Adam Danz 2019-1-10

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/439225-what-is-categorical-predictor-in-decision-tree-for-regression#answer_355905

编辑：Adam Danz 2019-1-10

Matlab's fitrtree() function returns a regression tree object. Read more about this object and its properties here:

https://www.mathworks.com/help/stats/classreg.learning.partition.regressionpartitionedmodel-class.html

As you'll read in the link above, the "CategoricalPredictors" contains index values corresponding to the columns of the categorical predictor data (if none of the predictors are categorical, this will be empty []).

So, why is it CategoricalPredictors equal to 2?

Now read about the function you're using fitrtree()

https://www.mathworks.com/help/stats/fitrtree.html#bt6cr84-tree

One of the name-value pairs (<- link) is 'CategoricalPredictors' which, is specified in your call to fitrtree() as 2. That's because you have two predictors being treated as categorical variables, [Weight, Cylinders].

3 个评论
显示 1更早的评论隐藏 1更早的评论

Salad Box 2019-1-15

在 MATLAB Online 中打开

Actually, as I read a few times, 'the "CategoricalPredictors" contains index values corresponding to the columns of the categorical predictor data', 'index value' (or the 'entry') means if index value is 1, that is the first column of the predictor data, in this case, it is 'Weight'; if 'index value' is 2, that is the second column of the predictor data, in this case, it is 'Cylinders'.

So

tree = fitrtree([Weight, Cylinders],MPG,...
                'categoricalpredictors',2,'MinParentSize',20,...
                'PredictorNames',{'W','C'})

indicates 'Cylinders' (the 2nd column in the predictor data) is a categorical predictor. In fact, there are only 4, 6, 8 cylinders, the number is not countinuous.

Rui An 2020-11-23