sklearn决策树完成卫星数据集分类

1、加载卫星数据集

sklearn决策树完成卫星数据集分类

2、分层采样划分训练集和测试集

使用交叉验证,所以不单独划分验证集

sklearn决策树完成卫星数据集分类

3、使用默认参数,交叉验证决策树的分类效果

sklearn决策树完成卫星数据集分类

4、网格搜索决策树最佳超参数

sklearn决策树完成卫星数据集分类

5、基于选择的最佳超参数,训练决策树模型,并在测试集上验证分类效果

sklearn决策树完成卫星数据集分类

分类准确率由默认参数的0.79提升到0.85