SVM, XGBoost, LightGBMの比較

タイトルの3つのモデルでAccuracy, 混同行列の比較を行いました。その結果、以下の出力が得られました。

SVM:
Accuracy: 0.8107037037037037
[[20283 723]
[ 4176 1818]]

XGBoost:
Accuracy: 0.8117777777777777
[[19772 1234]
[ 3611 2383]]

LightGBM:
Accuracy: 0.8095925925925925
[[19955 1051]
[ 3943 2051]]

3つのモデルで実際のテストデータでの解析を行いましたが、どれも82~83%のAccuracyで同じくらいの性能になりました。ソースコードは以下のリンクに置きました。
多少は性能上がるかなと思ったら、意外にも結果は同じだったようです。
https://github.com/TANEO-bio/Credit_svm_xgb_lgbm

Favicon
new user
コメントするには 新規登録 もしくは ログイン が必要です。