Stratified K-fold validation using crossval funtion

4 次查看（过去 30 天）
Jhon Gray
Jhon Gray 2021-4-20
回答： Rahul 2025-3-12
I am trying to customize the "Weighted kNN"-based classification code generated after a 10-fold cross validation on my data using the Classification Learner App (Using the Generate Function option in the app).
Query 1:
partitionedModel = crossval(trainedClassifier.ClassificationKNN, 'KFold',10);
After generating the code, I found that the above code is responsible for partitioning the data. I have an imbalanced dataset and I believe that "Stratified" k-fold cross validation ensures equal amount of data is considered from each class.
Query: Does the above code automatically perform Stratified k-fold cross validation? If not, how do I include it in the code?

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

回答（1 个）

Rahul
Rahul 2025-3-12
Hi @Jhon Gray,
You correctly identified 'crossval' function to be responsible for partitioning the data. Considering you are applying 'KFold' cross validation to your data, the 'crossval' function would stratify the data by default i.e. ensure equal amount of data is considered from each class.
For other categories of cross-validation tasks based on classification or if a particular class or list of class labels need to be specified to be specifically stratified, the 'Stratify' property of the 'crossval' function needs to be adjusted apprpriately.
The following MATLAB Answer can be referred:
https://www.mathworks.com/matlabcentral/answers/271843-how-does-crossval-for-k-fold-cv-work-in-matlab-after-training-a-classifier
The following MathWorks documentations can be referred to know more:
'Stratify' (crossval): https://www.mathworks.com/help/stats/crossval.html#mw_35debc35-cab8-4ad9-a97d-b039ac03a22b
Thanks.

类别

在 Help Center 和 File Exchange 中查找有关 Classification Learner App 的更多信息

标签

产品


版本

R2020a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by