naivebayes_usage.rb

require "naivebayes"

# training
nb = Naivebayes.new
datas = %w[ローズキングダム ウオッカ スクリーンヒーロー アドマイヤムーン ディープインパクト
アルカセット ゼンノロブロイ タップダンスシチー ファルブラヴ ジャングルポケット
テイエムオペラオー スペシャルウィーク エルコンドルパサー ピルサドスキー シングスピール
ランド マーベラスクラウン レガシーワールド トウカイテイオー ゴールデンフェザント
ベタールースンアップ ホーリックス ペイザバトラー ルグロリュー ジュピターアイランド
シンボリルドルフ カツラギエース スタネーラ ハーフアイスト メアジードーツ
]
datas.each do |e|
  nb.train(e, "ジャパンカップ１着")
end

# check
%w[
ジャガーメイル ブエナビスタ ローズキングダム オウケンブルースリ トレイルブレイザー
トゥザグローリー ペルーサ ヴィクトワールピサ サラリンクス キングトップガン
ミッションアプルーヴド ウインバリアシオン デインドリーム シャレータ
エイシンフラッシュ トーセンジョーダン
].each do |e|
  p = nb.docprob(e, "ジャパンカップ１着")
  puts "#{e}:" + sprintf("%0.1f",(p * 100)) + "%"
end