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

1、加載衛星數據集 2、分層採樣劃分訓練集和測試集 使用交叉驗證,所以不單獨劃分驗證集 3、使用默認參數,交叉驗證決策樹的分類效果 4、網格搜索決策樹最佳超參數 5、基於選擇的最佳超參數,訓練決策樹模型,並在測試集上驗證分類效果 分類準確率由默認參數的0.79提升到0.85
相關文章
相關標籤/搜索