Matlab桌上的K折交叉验证

时间:2018-11-07 18:03:55

标签: matlab logistic-regression cross-validation categorical-data naivebayes

我有一个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中很好地处理分类变量,因此按原样使用我的表及其分类变量应该没有问题。

但是,当我想使用cvpartitioncrossvalind进行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')

请问有人对此有看法吗?我知道我可以将分类变量更改为数字变量,但我宁愿不这样做。有没有办法解决?谢谢。

1 个答案:

答案 0 :(得分:0)

我认为您的主要问题是您的数据集太小。您有n = 5,这甚至不足以创建未经验证的模型。