[Ruby] 集合知プログラミング 03

haga — Sun, 19 Oct 2008 14:44:11 +0000

オライリー「集合知プログラミング」のPythonサンプルコードをRubyに翻訳していく作業ログ。３回目。

2.3.4 評者をランキングするのサンプルコードをRuby化

def top_matches( prefs, person, n=5, similarity=:sim_pearson)
scores = prefs.keys.select{|i| i != person}.map do |other|
[ method(similarity).call(prefs,person,other), other ]
end
return scores.sort.reverse[0...n]
end

p top_matches( critics, 'Toby', 3 )
# => [[0.99124070716193, "Lisa Rose"], [0.924473451641905, "Mick LaSalle"], [0.66284898035987, "Claudia Puig"]]

2.4 アイテムを推薦するのサンプルコードをRuby化

def get_recommendations( prefs, person, similarity=:sim_pearson )
totals = {}
sim_sums = {}
prefs.keys.each do |other|

# 自分自身とは比較しない
next if other == person
sim = method(similarity).call( prefs, person, other )
# 0以下のスコアは無視
next if sim <= 0

prefs[other].each do |item|
item_name = item[0]
# まだ見てない映画の得点のみ算出
if !prefs[person].key?(item_name) or prefs[person][item_name] == 0
# 類似度 * スコア
totals[item_name] ||= 0
totals[item_name] += prefs[other][item_name] * sim
# 類似度を合計
sim_sums[item_name] ||= 0
sim_sums[item_name] += sim
end
end
end

# 正規化したリストを作る
rankings = []
totals.each do |item_name,total|
rankings << [ total / sim_sums[item_name], item_name ]
end

# ソート済みのリストを返す
return rankings.sort.reverse

end

p get_recommendations( critics, 'Toby' )
# => [[3.27445961524987, "The Night Listener"], [2.95467531335197, "Lady in the Water"], [2.34855517312818, "Just My Luck"]]

引数similarityで、使用する類似性関数を切り替える。Rubyでは

method(:method_name).call

というような形で、シンボルまたは文字列でメソッド名を指定することでメソッドをコールできます。 http://www.ruby-lang.org/ja/man/html/Object.html#method

オリジナルのPythonのサンプルコードではタプルオブジェクトが多用されているが、Rubyでは配列に置き換えている。

[Ruby] 集合知プログラミング 02

haga — Wed, 15 Oct 2008 16:15:24 +0000

オライリー「集合知プログラミング」のピアソン相関によるスコア計算のサンプルプログラムをRubyに置き換えたもの

def sim_pearson( prefs, person1, person2 )

# 両者が互いに評価しているアイテムのリストを取得
si = prefs[person1].keys & prefs[person2].keys

# 要素の数を調べる
n = si.size

# 共に評価しているアイテムがなければ0を返す
return 0 if n == 0

# 配列をブロックで収集し、合計するメソッド
def map_sum( list, &block )
list.map{|i| yield(i) }.inject(0.0){|res,item| res += item }
end

# すべての嗜好を合計する
sum1 = map_sum( si ){|i| prefs[person1][i] }
sum2 = map_sum( si ){|i| prefs[person2][i] }

# 平方を合計する
sum1_sq = map_sum( si ){|i| prefs[person1][i] ** 2 }
sum2_sq = map_sum( si ){|i| prefs[person2][i] ** 2 }

# 積を合計する
p_sum = map_sum(si){|i| prefs[person1][i] * prefs[person2][i] }

# ピアソンによるスコアを計算する
num = p_sum - ( sum1 * sum2 / n)
den = Math.sqrt(( sum1_sq - sum1 ** 2 / n ) * ( sum2_sq - sum2 ** 2 / n ))
return 0 if den == 0

return num / den

end

puts sim_pearson( critics, 'Lisa Rose', 'Gene Seymour' )

pythonでリスト内包を使っての合計は、Rubyで書くとmapとinjectを組み合わせたものになると思う。これはPythonに比べ長くなってしまい、読みにくかった上に繰り返し出現するのでmap_sumというショートカット用のメソッドを用意した。

# 長ったらしい
sum1 = si.map{|i| prefs[person1][i] }.inject(0.0){|res,item| res += item }

# python版
sum1 = sum([prefs[p1][it] for it in si])

Ruby版は式を評価する順番が、単純に左から右へ読めばいいのに対し、Python版では式の評価箇所が飛び飛びなのが、Pythonがわからない人間には辛いが、慣れればリスト内包の方が奇麗な気がしてくるから不思議だ。

両者の評価しているアイテムを収集する場合は、Hash#keys同士を&演算子で、集合の積を抽出するのがRubyらしいのかなと思う。

# 両者が互いに評価しているアイテムのリストを取得
si = prefs[person1].keys & prefs[person2].keys

func09 » 集合知

[Ruby] 集合知プログラミング 03

[Ruby] 集合知プログラミング 02