コンテンツへスキップ
FindOpenSource

XGBoost

構造化・表形式データの問題に広く使われている、最適化された分散型勾配ブースティングライブラリで、応用機械学習の定番です。

概要

XGBoost は、パフォーマンスとスケーラビリティを強く重視した勾配ブースティング決定木を実装しており、構造化(表形式)データ——画像やテキストではなく、スプレッドシートやデータベースにあるようなデータ——を対象とする機械学習コンペティションにおいて、最も安定して勝ち続けているアルゴリズムの1つです。単一マシンでも、クラスタに分散させても効率的に動作します。

XGBoost は、構造化データに対する分類、回帰、ランキングの問題を解くチームに向いています。画像・テキスト・音声のタスクを席巻しているものの、表形式データでは勾配ブースティングに性能で劣ることが多い、(カタログにもある) 深層学習フレームワーク(PyTorch、TensorFlow)とは異なるツールです。

カテゴリ
AI・機械学習
キーワード
gradient-boostingmachine-learningtabular-dataclassification
言語
C++
ライセンス
Apache-2.0

誤りを見つけましたか? GitHub で修正を提案する