adaboost 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c0f5373579f8f93316192361c02c0b29e176f09b
-  data.tar.gz: 6f1c7a127b3b2283fab1b2bf3b4679e32e4587b7
+  metadata.gz: f550adc8429f2927416f49e86461497d9f43a072
+  data.tar.gz: bf2fa99dad4d438a25588f93d8de7a2c62c7f45f
 SHA512:
-  metadata.gz: 28f2c7b14a2e372be409a593c0ec45fda9feedbb08d700ae85b2714dbea3121aba8660b271711a4dac67245293fed1c579cebc662040dfc6d89e6db7d85f9c84
-  data.tar.gz: 6ea5ebf0d20243bbb5ba409990d3f7950d842a6a713ee6f14169ab96832e95d7b0296c884c63322f9c5066fc3511387fc775350fc1e63102625af2aa848734ed
+  metadata.gz: 38b3488dd4d034e9694200841a8b1c5022b60f8e8cb06afd88b99147dc140b73183153b12d7ba5c79e06865969958051dee9b59ef76af91e2901731626681b6b
+  data.tar.gz: 8f41f660c6a53b66ee8eed9a80b3948a66bae0bb516ab8d4f52d3322a66c0f40f9201f700cb499be885c5727bc1d2b65966efd04f48fe4bfdcb4453c8607fed9

data/lib/adaboost/adaboost.rb

@@ -23,6 +23,7 @@ module AdaBoost
         weak_classifier.compute_alpha
         update_weights weak_classifier, samples
         @weak_classifiers << weak_classifier
+        yield i, weak_classifier if block_given? 
       end
     end
 
@@ -50,21 +51,21 @@ module AdaBoost
       negative_weight = 1 / samples_size
       positive_weight = negative_weight
       if Config::INCORPORATE_COST_SENSITIVE_LEARNING
-          analyzer = FeaturesAnalyzer.new @y_index
-          distribution = analyzer.analyze(samples).distribution
-          positive_rate = distribution.positive / samples_size
-          negative_rate = distribution.negative / samples_size
-          normalizing_constant = distribution.negative * positive_rate + distribution.positive * negative_rate
-          positive_weight = positive_rate / normalizing_constant.to_f
-          negative_weight = negative_rate / normalizing_constant.to_f
+        analyzer = FeaturesAnalyzer.new @y_index
+        distribution = analyzer.analyze(samples).distribution
+        positive_rate = distribution.positive / samples_size
+        negative_rate = distribution.negative / samples_size
+        normalizing_constant = distribution.negative * positive_rate + distribution.positive * negative_rate
+        positive_weight = positive_rate / normalizing_constant.to_f
+        negative_weight = negative_rate / normalizing_constant.to_f
       end
       samples.each_with_index do |sample, i|
-          y = sample[@y_index]
-          if y == -1 
-              @weights[i] = positive_weight
-          else 
-              @weights[i] = negative_weight
-          end
+        y = sample[@y_index]
+        if y == -1 
+          @weights[i] = positive_weight
+        else 
+          @weights[i] = negative_weight
+        end
       end
     end
 
@@ -76,7 +77,7 @@ module AdaBoost
         sum += @weights[i]
       end
       @weights.each_with_index do |_, i|
-          @weights[i] /= sum
+        @weights[i] /= sum
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: adaboost
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Dalmir da Silva