require 'ostruct'
+require 'logger'
+$log = Logger.new(STDERR)
+$log.level = Logger::WARN
+
# A simple graph library
module GraphNjae
iteration = 1
residual = max_residual + 1
while residual > max_residual and iteration <= max_iterations
-# puts "Starting iteration #{iteration}"
+ $log.debug { "Starting iteration #{iteration}" }
self.vertices.each do |v|
v.last_similarity = v.similarity
end
self.vertices.each do |v|
-# puts "Processing vertex #{v.name}"
- edge_groups = v.edges.group_by {|e| e.type }
-# puts " Edge groups {#{edge_groups.keys.map {|t| t.to_s + ' => {' + edge_groups[t].map {|e| e.to_s}.join(', ')}.join('; ')}}"
- edge_groups.each do |type, edges|
-# puts " Processing group type #{type}"
- n = edges.length
- edges.each do |e|
- e.other_end(v).similarity += v.last_similarity / n
+ if block_given?
+ v.similarity = yield v
+ else
+ $log.debug { "Processing vertex #{v.name}" }
+ edge_groups = v.edges.group_by {|e| e.type }
+ $log.debug { " Edge groups {#{edge_groups.keys.map {|t| t.to_s + ' => {' + edge_groups[t].map {|e| e.to_s}.join(', ')}.join('; ')}}" }
+ edge_groups.each do |type, edges|
+ $log.debug { " Processing group type #{type}" }
+ n = edges.length
+ edges.each do |e|
+ e.other_end(v).similarity += v.last_similarity / n
+ end
end
end
end
-# self.vertices.each do |v|
-# n = v.neighbours.length
-# v.neighbours.each do |neighbour|
-# neighbour.similarity += v.last_similarity / n
-# end
-# end
max_similarity = vertices.map {|v| v.similarity}.max
self.vertices.each do |v|
v.similarity = v.similarity / max_similarity
end
residual = Math.sqrt(self.vertices.reduce(0) {|a, v| a += (v.similarity - v.last_similarity) ** 2})
-# puts "Residual = #{residual.round(3)}, sims = #{self.vertices.map {|v| v.name + " = " + v.similarity.round(2).to_s}}"
+ $log.debug { puts "Residual = #{residual.round(3)}, sims = #{self.vertices.map {|v| v.name + " = " + v.similarity.round(2).to_s}}" }
iteration += 1
end