我有一个Matlab表,其中包含有关学生的信息(数字和分类)。此处提供了一个示例:
School = {'GB'; 'UR'; 'GB'; 'GB'; 'UR'};
School = categorical(School);
Age = [14;14;12;16;19];
Relationship = {'yes'; 'yes'; 'no'; 'no'; 'yes'};
Relationship = categorical(Relationship);
Status = {'ft'; 'pt'; 'ft'; 'ft'; 'ft'};
Status = categorical(Status);
Father_Job = {'pol'; 'ser'; 'oth'; 'ele'; 'cle'};
Father_Job = categorical(Father_Job);
Health = [1;2;3;3;5];
Exam = {'pass'; 'pass'; 'fail'; 'fail'; 'pass'};
Exam = categorical(Exam);
T =
School Age Relationship Status Father_Job Health Exam
______ ___ ____________ ______ __________ ______ ____
GB 14 yes ft pol 1 pass
UR 14 yes pt ser 2 pass
GB 12 no ft oth 3 fail
GB 16 no ft ele 3 fail
UR 19 yes ft cle 5 pass
我想使用这些数据来预测和分类考试的通过/失败。我打算使用fitglm
进行逻辑回归,并使用fitcnb
进行朴素贝叶斯分类。我知道这两种方法都可以在Matlab中很好地处理分类变量,因此按原样使用我的表及其分类变量应该没有问题。
但是,当我想使用cvpartition
和crossvalind
进行10倍交叉验证时遇到问题。当我尝试创建折痕索引时,出现以下错误:使用statslib.internal.grp2idx的错误不支持使用线性索引(一个下标)或多维索引(三个或多个下标)对表进行下标。使用行下标和变量下标。
我的目标是执行以下操作:
% Column 7 (Exam) is the response variable
X = T(:, 1:6);
Y = T(:, 7);
% Create indices of 5-fold cross-validation (here I get errors)
cvpart = cvpartition(Y,'KFold',5);
indices = crossvalind('Kfold',Y,5);
% Create my test and training sets
for i = 1:5
test = (indices == i);
train = ~test;
Xtrain = X(train,:);
Xtest = X(test,:);
Ytrain = Y(train,:);
Ytest = Y(test,:);
end
% Fit logistic model
mdl = fitglm(Xtrain,Ytrain,'Distribution','binomial')
请问有人对此有看法吗?我知道我可以将分类变量更改为数字变量,但我宁愿不这样做。有没有办法解决?谢谢。
答案 0 :(得分:0)
我认为您的主要问题是您的数据集太小。您有n = 5,这甚至不足以创建未经验证的模型。