matlab 随机森林_基于机器学习的心律失常分类(六)——随机森林分类[MATLAB]

一、随机森林

二、代码

说明：

1. DATA矩阵为arma系数矩阵，1-6列为ARMA模型系数，第七列为所数类别（1-正常心电，2-左束支阻滞，3-右束支阻滞，4-室性早搏）
2. 四个类别分别选取各1700个数据，共计6800个样本，取70%作为训练集。

%% I. 清空环境变量

三、结果分析

关于评估指标的含义在《基于机器学习的心律失常分类(五)——决策树分类》里已经说过了，这里就直接给出分类结果。

本文数据包括四类不同的心律波形，共计6800个样本心拍数据，随机提取其中的70%(4760个)为训练样本，30%(2040个)为测试样本。在学习训练样本，通过测试样本进行分类验证，分别得到分类模型的混淆矩阵，竖直方向表示预测样本的标签类别，水平方向表示真实样本的标签类别。

随机森林混淆矩阵

随机森林评估指标