Changeset 34106 for lang/ruby/ruby-bayon/trunk/lib

Timestamp:

06/24/09 01:44:46 (4 years ago)

Author:

winebarrel

Message:

bayon 0.0.4にあわせて類似度を表示するオプションを追加

Files:

• lang/ruby/ruby-bayon/trunk/lib/bayon.rb (modified) (4 diffs)

lang/ruby/ruby-bayon/trunk/lib/bayon.rb

@@ -7 +7 @@
       @cluster_size_limit = nil
       @eval_limit = nil
+      @output_similairty_point = nil
     end
 
@@ -25 +26 @@
     end
 
+    def output_similairty_point=(output)
+      unless output.instance_of?(TrueClass) or output.instance_of?(FalseClass)
+        raise TypeError, "wrong argument type #{limit.class} (expected boolean value)"
+      end
+
+      @output_similairty_point = output
+    end
+    
     def add_document(label, features)
       unless features.kind_of?(Hash)
@@ -41 +50 @@
       analyzer.set_cluster_size_limit(@cluster_size_limit) if @cluster_size_limit
       analyzer.set_eval_limit(@eval_limit) if @eval_limit
+      analyzer.set_output_similairty_point(@output_similairty_point) if @output_similairty_point
 
       feature_set = []
@@ -60 +70 @@
 
       while (cluster = analyzer.get_next_result)
-        result << cluster.map {|doc_id| @documents[doc_id][0] }
+        if @output_similairty_point
+          result << cluster.map {|doc_id, point| [@documents[doc_id][0], point] }
+        else
+          result << cluster.map {|doc_id| @documents[doc_id][0] }
+        end
       end