Show
Ignore:
Timestamp:
06/24/09 01:44:46 (6 years ago)
Author:
winebarrel
Message:

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

Files:
1 modified

Legend:

Unmodified
Added
Removed
  • lang/ruby/ruby-bayon/trunk/lib/bayon.rb

    r33943 r34106  
    77      @cluster_size_limit = nil 
    88      @eval_limit = nil 
     9      @output_similairty_point = nil 
    910    end 
    1011 
     
    2526    end 
    2627 
     28    def output_similairty_point=(output) 
     29      unless output.instance_of?(TrueClass) or output.instance_of?(FalseClass) 
     30        raise TypeError, "wrong argument type #{limit.class} (expected boolean value)" 
     31      end 
     32 
     33      @output_similairty_point = output 
     34    end 
     35     
    2736    def add_document(label, features) 
    2837      unless features.kind_of?(Hash) 
     
    4150      analyzer.set_cluster_size_limit(@cluster_size_limit) if @cluster_size_limit 
    4251      analyzer.set_eval_limit(@eval_limit) if @eval_limit 
     52      analyzer.set_output_similairty_point(@output_similairty_point) if @output_similairty_point 
    4353 
    4454      feature_set = [] 
     
    6070 
    6171      while (cluster = analyzer.get_next_result) 
    62         result << cluster.map {|doc_id| @documents[doc_id][0] } 
     72        if @output_similairty_point 
     73          result << cluster.map {|doc_id, point| [@documents[doc_id][0], point] } 
     74        else 
     75          result << cluster.map {|doc_id| @documents[doc_id][0] } 
     76        end 
    6377      end 
    6478