Multi Variable Categorical Regression

Question

Connor Stepkoski 2020-4-20

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/519375-multi-variable-categorical-regression

关闭： MATLAB Answer Bot 2021-8-20

BaseballData.xlsx

I'm attempting to build a predictor for attendence at baseball games. Based on 5 variables, 4 of which are categorical. I've tried a multitude of different things and am having no luck. Does anyone have any advice/ideas on how I should go about this?

I've attatched the data I've compsed.

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

the cyclist 2020-4-20

It would be helpful to know ...

what sorts of things you tried (e.g. which MATLAB functions?)
what does "having no luck" mean? Do you have code that crashes? Code that runs, but doesn't result in a "good" model?
whether or not you have the Statistics and Machine Learning Toolbox
a little bit of info on your level of experience with MATLAB
a little bit of info on your level of experience using predictive models (either in MATLAB or other language)

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

Answer 1

the cyclist 2020-4-20

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/519375-multi-variable-categorical-regression#answer_427167

编辑：the cyclist 2020-4-20

在 MATLAB Online 中打开

The following code fits a binary regression tree to your data:

baseballData = readtable('BaseballData.xlsx'); % Load the data into a table
baseballData.Date = []; % Remove the date, which presumably should not be used for prediction
tree = fitrtree(baseballData,'Attendence'); % Fit the model

You can then predict attendance for a particular set of parameters. Here, I predict attendance for the first row of the table:

predict(tree,baseballData(1,:))

You could also do a simple linear model using

linear = fitlm(baseballData)

Note that for both of these models, the syntax is particularly simple because your response variable happens to be the last variable in the table (which is the default assumption by MATLAB), and because the functions are able to figure out automatically which variables are categorical.