opener-scorer 3.0.2 → 3.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3c4de6d63c9229258f32afd1ee2b74f664c2209c
-  data.tar.gz: ded5b0be3d2715f56b3a3e5ee41e24058549109a
+  metadata.gz: 1dcd08283a4bd6b1d6e99e8925bb738820f529cf
+  data.tar.gz: 4212ffb0e1a53ff57452745d858f304b8d218f85
 SHA512:
-  metadata.gz: ff27e75c29affc0264ea1507f2a9223c3682c4f53064bd275edb460ae68c6760fc41a7c46f886283b05e64a12845f82091933e6e09c49c8c4c49114ab9496638
-  data.tar.gz: f7e36e7640d00c525583c0805fbaa0c5ac97b5a343175466a081d8160b3b0f1cab5b0650369452a488c87d26ccbf0e80333b07f4b04a99eaebe4d9fb1b5f8ba8
+  metadata.gz: 466ce5a7591c03b4d3a15e2c06992ff8df5b07925485aab7797864fd1b0a3e116cdd0c2438bb5ce44bdeb051239adee5a831f504050b0eb8ba88537d129abc53
+  data.tar.gz: 9c5cf6c073ccebe565c0b379bcc53a8cfb62b60a8b61a379861abd580fbb95646d36232c477067012b17f8685c6044f2ea5ef70b3c94d310f0a6dd9d68f1fa07

data/lib/opener/scorer/output_processor.rb

@@ -85,32 +85,33 @@ module Opener
         return lemmas_hash
       end
 
-      ##
-      # Create a hash with all lemma ids that have a polarity.
-      #
       # @param [Nokogiri::XML::Document] document
       # @return [Hash]
       #
       def build_polarities_hash(document)
-        polarities_hash = Hash.new { |hash, key| hash[key] = [] }
+        polarities_hash = {}
         opinions        = document.at('opinions')
 
         return polarities_hash unless opinions
 
         opinions.css('opinion').each do |opinion|
           polarity  = opinion.at('opinion_expression').attr('polarity').to_sym
+          strength = opinion.at('opinion_expression').attr('strength').to_i.abs
           op_target = opinion.at('opinion_target')
           op_expr   = opinion.at('opinion_expression')
 
           if op_target
             op_target.css('span target').each do |target|
-              polarities_hash[target.attr('id')] << polarity
+              require 'pry'
+              polarities_hash[target.attr('id')] ||= {} rescue binding.pry
+              polarities_hash[target.attr('id')][polarity] = strength
             end
           end
 
           if op_expr
-            op_expr.css('span target').each do |target|
-              polarities_hash[target.attr('id')] << polarity
+            op_expr.css('span target').each do |expression|
+              polarities_hash[expression.attr('id')] ||= {}
+              polarities_hash[expression.attr('id')][polarity] = strength
             end
           end
         end
@@ -125,17 +126,20 @@ module Opener
       # @return [Float]
       #
       def get_overall_score(document)
-        polarities = []
+        polarities = {}
+        polarities[:positive] = []
+        polarities[:negative] = []
         opinions   = document.at('opinions')
 
         return 0.0 unless opinions
 
         opinions.css('opinion').each do |opinion|
-          polarities << opinion.at('opinion_expression').attr('polarity').to_sym
+          sentiment = opinion.at('opinion_expression').attr('polarity').to_sym
+          polarities[sentiment] << opinion.at('opinion_expression').attr('strength').to_i.abs
         end
 
-        positive = polarities.count(:positive)
-        negative = polarities.count(:negative)
+        positive = polarities[:positive].inject(0, :+)
+        negative = polarities[:negative].inject(0, :+)
 
         return if (positive + negative) == 0
 
@@ -166,15 +170,17 @@ module Opener
       # @return [Float]
       #
       def calculate_score(lemma_ids, polarities_hash)
-        polarities = []
+        positive_polarities = []
+        negative_polarities = []
 
         lemma_ids.each do |id|
-          polarities << polarities_hash[id]
+          positive_polarities << polarities_hash[id].fetch(:positive, 0)
+          negative_polarities << polarities_hash[id].fetch(:negative, 0)
         end
 
-        positive = polarities.flatten.count(:positive)
-        negative = polarities.flatten.count(:negative)
-
+        positive = positive_polarities.compact.inject(0, :+)
+        negative = negative_polarities.compact.inject(0, :+)
+        
         return if (positive + negative) == 0
 
         score = ((positive - negative).to_f) / (positive + negative)

data/lib/opener/scorer/version.rb

@@ -1,5 +1,5 @@
 module Opener
   class Scorer
-    VERSION = '3.0.2'
+    VERSION = '3.0.3'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: opener-scorer
 version: !ruby/object:Gem::Version
-  version: 3.0.2
+  version: 3.0.3
 platform: ruby
 authors:
 - development@olery.com
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2014-12-03 00:00:00.000000000 Z
+date: 2014-12-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: nokogiri
@@ -181,28 +181,28 @@ dependencies:
 description: Component for calculating scores of KAF documents
 email:
 executables:
-- scorer-server
-- scorer-daemon
 - scorer
+- scorer-daemon
+- scorer-server
 extensions: []
 extra_rdoc_files: []
 files:
 - config/database.rb
 - lib/opener/scorer.rb
-- lib/opener/scorer/version.rb
-- lib/opener/scorer/output.rb
-- lib/opener/scorer/server.rb
 - lib/opener/scorer/cli.rb
+- lib/opener/scorer/output.rb
 - lib/opener/scorer/output_processor.rb
+- lib/opener/scorer/server.rb
+- lib/opener/scorer/version.rb
 - lib/opener/scorer/views/index.erb
 - config.ru
 - opener-scorer.gemspec
 - README.md
 - LICENSE.txt
 - exec/scorer.rb
-- bin/scorer-server
-- bin/scorer-daemon
 - bin/scorer
+- bin/scorer-daemon
+- bin/scorer-server
 homepage: http://opener-project.github.com/
 licenses:
 - Apache 2.0
@@ -228,4 +228,3 @@ signing_key:
 specification_version: 4
 summary: Component for calculating scores of KAF documents
 test_files: []
-has_rdoc: yard