回帰と分類。線形モデル・決定木の勾配ブースティング・ニューラルネットの使い分け。
教師あり学習は、入力と正解(ラベル)の組から「入力 → 答え」の対応を学ぶ方法です。答えが数値なら回帰(例:来月の売上)、種類なら分類(例:不正か正常か)です。代表的な手法は次の3つで、データの形によって向き不向きがはっきりしています。
| 手法 | 得意なデータ | 長所 | 注意点 |
|---|---|---|---|
| 線形モデル(線形回帰・ロジスティック回帰) | 特徴量が少なく関係が単純なもの | 速い・係数で説明しやすい・過学習しにくい | 組み合わせ(交互作用)や非線形は自分で特徴量にする必要がある |
| 勾配ブースティング決定木(LightGBM・XGBoost・CatBoost) | 表形式のデータ(数値とカテゴリの列が並ぶもの) | 前処理が少なくても強い・欠損値を扱える・学習が速い | 外挿(学習した範囲の外)は苦手。木の数・深さで過学習する |
| ニューラルネット | 画像・音声・文章など構造のあるデータ | 大量のデータで表現そのものを学べる | データと計算が多く必要。表形式では勾配ブースティングに負けることも多い |
回帰か分類かは、答えの種類で決まります。同じ「解約」でも、「解約するか」は分類、「何日後に解約するか」は回帰(または生存時間の分析)です。
面接ではこう答える
教師あり学習は、入力と正解の組から「入力から答え」への対応を学ぶ方法で、答えが数値なら回帰、種類なら分類です。手法はデータの形で選びます。表形式のデータなら勾配ブースティング決定木がまず強く、画像や文章のような構造のあるデータはニューラルネット、特徴量が少なく説明が必要なら線形モデルを使います。いずれの場合も、まず単純なベースラインを作ってから比べます。
MLとの関わり「表形式のデータでまず何を試すか」は面接の定番で、実務でもモデル選びの最初の判断になります。