今回はベイズの定理という数学(確率論)の定理をもとにした、Naive Bayes アルゴリズムと呼ばれる教師あり学習アルゴリズムについて説明します。 これは分類タスク向けのアルゴリズムで、文書データに対してよい結果を出すことが知られています。 用意してある、Jupyter notebookをダウンロードして実際にコードを書きながら学習することができます。 今回は、下記にコード例も最後に載せてあるので、ぜひ自分の手で実行して理解を深めてみてください。
機械学習のタスクでは、データセットが与えられたときに、それをもとにした推定がいくつかある中で、どの推定が最もらしいかということがよく問題になります。
例えば分類タスクでは、そのデータ点(サンプル)がどれに分類されるのかというのが、ここでいう推定ということになります。
このような、どの推定が最もらしいか判断するときに使えるのが、ベイズの定理と呼ばれる定理です。
ベイズの定理の式
[math] \underbrace{P(H|D)}_{結果が起きた時の原因の確率} = \underbrace{ \frac {P(D|H) \, P(H)}{P(D)}}_{各値はデータから計算できる値} [/math]
[math] P(Fire|Smoke) = \frac{P(Fire)\, P(Smoke|Fire)} {P(Smoke)} = \frac{0.01 * 0.9}{0.1} = 0.09 = 9% [/math]
ということになります。
煙が見えた時に火事が起こっている(家事が煙の原因である)確率の推定結果
このようにして各原因の推定が正しい確率を計算していきます。
ベイズの定理をもとにしたのがNaive Bayesアルゴリズムであり、以下のような特徴を持ちます。
以下にNaive Bayesアルゴリズムの長所と短所をまとめます。
| 長所 |
|
| 短所 |
|
そして以下に応用先をまとめます。 上であげた長所から、リアルタイムでの処理や文書データによく用いられています。
実用例として、電車の混雑状況や、津波・地震などの災害予測、株価のリアルタイム予測などがあり、リアルタイムで状況を予測する機能です。
テキスト分類とは、与えられた文書（Webページなど）をあらかじめ与えられたいくつかのカテゴリ（クラス）に自動分類することです。テキスト分類は、対象とするテキストによって幅広い応用ができます。例えば、すでに実用化されている機能としては、
受診したメールを「スパム」と「それ以外」というカテゴリへ自動分類して「スパム」をゴミ箱へ捨てる、スパムフィルタ機能。
「政治・経済」「科学・学問」「コンピュータ・IT」「ゲーム・アニメ」などのカテゴリへ自動分類（はてなブックマーク）。
記事を、「興味あり」「興味なし」というカテゴリへ自動分類して「興味あり」のニュース記事だけおすすめ（情報推薦・情報フィルタリング）に表示。 などがあります。
例えば、よくネットショッピングをしている時に見かける、「あなたの検索したアイテムを買った人は、このアイテムも買っています」という表現です。 それまでの利用者の行動を蓄積し、機械が学習することで、あなたに合うと思われるアイテムなどを勧めてくれます。
センチメント分析とは、あるトピックに対して、その文章が肯定的であるか否定的であるか、あるいは中立的であるかを判断するものです。特定のテーマに対して、人々がどのように感じているかを分析する際に用いられます。
ここでは実際にScikit learnで実装したコード例を載せておきます。ぜひ実際に実行して、その手軽さを実感してみてください。 Naive Bayesの中には実はいくつかタイプがあるのですが、ここではGaussianモデルと呼ばれるものを用いています。 内容としては、自分で用意した2次元のトレーニングデータセットをもとに、新しい2次元データをlabel1かlabel2に分類するというシンプルなものです。 実際の実行は他の機械学習Academy同様、こちらのJupyter Notebookをダウンロードしてその中で行ってください。 ＊Jupyter Notebook使用は、Anacondaがダウンロード済みである必要があります。 Anacondaダウンロード方法は機械学習を用いた画像分類体験をご覧ください。
まずは必要なライブラリをインポートします。 Scikit learnからGaussianNB(Gaussian型Naive Bayes)を呼んでいます。
|
1 2 3 4 5 |
# 必要なライブラリのインポート from sklearn.naive_bayes import GaussianNB import numpy as np import matplotlib.pyplot as plt %matplotlib inline |
2次元のトレーニングデータを自前で用意します。 xがデータ、yがそのlabel(1 or 2)になります。
|
1 |
#トレーニングデータセットの用意 x = np.array([[-2,6], [0,6], [0,7], [-2,5], [-3,3], [-1,0], [-2,0], [-3,1], [-1,4], [0,3], [0,1], [-1,7], [-3,5], [-4,3], [-2,0], [-3,7], [1,5], [1,2], [-2,3], [2,3], [-4,0], [-1,3], [1,1], [-2,2], [2,7], [-4,1]]) y = np.array([2, 1, 2, 1, 1, 2, 2, 2, 1, 1, 2, 2, 2, 2, 2, 1, 1, 1, 2, 2, 1, 1, 2, 1, 2, 2]) |
用意したトレーニングデータを2次元平面に描画して、可視化してみましょう。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# グラフ描画に向けてのデータの整形 data = np.hstack((x, y.reshape(y.shape[0],1))) # print(data) data1 = data[np.where(data[:,2]==1)] data2 = data[np.where(data[:,2]==2)] # print(data1) # print(data2) # matplotlibを用いたデータの可視化（グラフ化） plt.close("all") plt.scatter(data1[:,0], data1[:,1], c="tab:blue") plt.scatter(data2[:,0], data2[:,1], c="tab:red") plt.show() |
多少の外れ値はありますが、全体として中心の青（label1）のデータ点の集まりを赤（label2）のデータが囲んでいるような構造が見えます。 それではこのトレーニングデータを用いて、Gaussian Naive Bayesを学習させます。 学習はほぼ一瞬で終わります。
|
1 2 3 4 |
# モデルの学習 model = GaussianNB() model.fit(x, y) print("Model fitted.") |
次にテストデータを用意し、それをグラフに描画してみましょう。
|
1 2 3 4 5 6 7 8 9 |
# テストデータの用意 test_data = np.array([[0,4], [1,0]]) # matplotlibを用いたデータの可視化（グラフ化） plt.close("all") plt.scatter(data1[:,0], data1[:,1], c="tab:blue") plt.scatter(data2[:,0], data2[:,1], c="tab:red") plt.scatter(test_data[:,0], test_data[:,1], c="k" ) plt.show() |
黒の点がテストデータ[0,4]と[1,0]になります。 [0,4]は青（label1）のデータ点の集まりの中に位置し、[1,0]は外側の赤（label2）のデータのあたりに位置しています。 それではテストデータを学習したモデルに与えて分類させてみます。
|
1 2 3 4 |
# テストデータの分類 test_label = model.predict(test_data) print("Label of test data", test_data[0], ":", test_label[0]) print("Label of test data", test_data[1], ":", test_label[1]) |
用意したテストデータ[0,4]はlabel1に、[1,0]はlabel2に分類されていることがわかります。 この結果から、青（label1）のデータの外側に赤（label2）のデータというトレーニングデータの構造を、モデルがうまく掴んでいるであろうことがわかります。 他にもトレーニングデータを変えてみたり、テストデータを変えてみたりして色々な設定で動かしてみてましょう。アルゴリズムの特徴をつかむのに役に立つはずです。
今回はベイズの定理という、数学(確率)の定理を用いた教師あり学習アルゴリズムの1つ、Naive Bayesについて、Example code付きで見てきました。 主な利用先は分類問題で、単純(実装も簡単)かつ大きなデータセットに対しても高速に計算ができるのが特徴です。 コードに関してはぜひ色々なデータを自分で設定してみて、アルゴリズムの特徴をつかんでみてください。
社員の成長を導きながら、AIやビッグデータなどの最先端技術をプロジェクトに活用していくことが私たちのビジョンです。Avintonの充実した技術研修でスキルアップを図り、あなたのキャリア目標を一緒に達成しませんか？
