sklearn決策樹完成衛星數據集分類

1、加載衛星數據集

2、分層採樣劃分訓練集和測試集

使用交叉驗證,所以不單獨劃分驗證集

3、使用默認參數,交叉驗證決策樹的分類效果

4、網格搜索決策樹最佳超參數

5、基於選擇的最佳超參數,訓練決策樹模型,並在測試集上驗證分類效果

分類準確率由默認參數的0.79提升到0.85

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章