Class: FeatureSelection::FrequencyBased

Inherits:
Base
  • Object
show all
Defined in:
lib/feature_selection/algorithms/frequency_based.rb

Instance Method Summary collapse

Methods inherited from Base

#classes, #initialize

Methods included from LogHelpers

#create_log, #log_calculations_complete, #write_to_log

Constructor Details

This class inherits a constructor from FeatureSelection::Base

Instance Method Details

#rank_features ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/feature_selection/algorithms/frequency_based.rb', line 4

def rank_features
  write_to_log("Starting to rank features...")
  # Returns:
  #=> {:class => {'term' => count, 'term' => count}}
  @results = {}
  
  # For logger
  total_calculations = classes.size * terms.size
  n = 1
  
  classes.each do |klass|
    @results[klass] = {}
    
    terms.each do |term|
      log_calculations_complete(n)
      n += 1
      
      if @results[klass].key?(term)
        @results[klass][term] += 1
      else
        @results[klass][term] = 1
      end
    end #terms.each
  end #classes.each
  @results
end