pearson 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e0bbce86ff859e971101cf43995637d7d56d07f6
+  data.tar.gz: 72e1811067f3d044213e55d1336d74a2c1163606
+SHA512:
+  metadata.gz: b4f9b2327f220781a24f3ab96f0087c1a7819b33a6fa866137d05773e932b83448bda278a8c9e202894f6a72c842386864a7ab10e62661bcf1b1eaf62643b7ed
+  data.tar.gz: d75ebd9be509a4aba48b2eb9d77574dc37aa46545c49cead8414f1b34cc187428a44d8068c525ad683f5e12b3604cfa5ee14df759dd08afce410165cc1419184

data/.ruby-version

@@ -0,0 +1 @@
+2.0.0-p247

data/Gemfile

@@ -0,0 +1,5 @@
+source 'http://rubygems.org'
+
+group :development do
+  gem 'rspec', '>= 2.14.1'
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Alfonso Jiménez
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/lib/pearson.rb

@@ -0,0 +1,108 @@
+module Pearson
+  class << self
+    # Calculates the pearson correlation coefficient between
+    # two entities
+    #
+    # @param [Hash] Hash containing entity-item scores
+    # @param [String] First entity
+    # @param [String] Second entity
+    #
+    # @return [Float] Coefficient
+    def coefficient(scores, entity1, entity2)
+      shared_items = shared_items(scores, entity1, entity2)
+
+      n = shared_items.length
+
+      return 0 if n == 0
+
+      sum1 = sum2 = sum1_sq = sum2_sq = psum = 0
+
+      shared_items.each_key do |item|
+        sum1 += scores[entity1][item]
+        sum2 += scores[entity2][item]
+
+        sum1_sq += scores[entity1][item]**2
+        sum2_sq += scores[entity2][item]**2
+
+        psum += scores[entity1][item]*scores[entity2][item]
+      end
+
+      num = psum - (sum1*sum2/n)
+      den = ((sum1_sq - (sum1**2)/n) * (sum2_sq - (sum2**2)/n)) ** 0.5
+
+      den == 0 ? 0 : num/den
+    end
+
+    # Returns the closest entities from a given entity. The distance
+    # between entities is based on the Pearson correlation coefficient
+    #
+    # @param [Hash] Hash containing entity-item scores
+    # @param [String] Entity
+    # @param [Hash] Options (limit)
+    #
+    # @return [Array] Top matches
+    def closest_entities(scores, entity, opts={})
+      sort_desc(scores, opts) do |h, k, v|
+        entity == k ? h : h.merge(k => coefficient(scores, entity, k))
+      end 
+    end
+
+    # Returns the best recommended items for a given entity
+    #
+    # @param [Hash] Hash containing entity-item scores
+    # @param [String] Entity
+    # @param [Hash] Options (limit)
+    #
+    # @return [Array] Top matches [item, score]
+    def recommendations(scores, person, opts={})
+      totals = {}
+      similaritySums = {}
+
+      totals.default = 0
+      similaritySums.default = 0
+
+      fail PersonNotFound unless scores[person]
+
+      scores.each do |other_person|
+        next if other_person.first == person
+
+        similarity = coefficient(scores, person, other_person.first)
+
+        next if similarity <= 0
+
+        scores[other_person.first].each do |item, score|
+          if !scores[person].keys.include?(item) || scores[person][item] == 0
+            totals[item] += score * similarity
+            similaritySums[item] += similarity
+          end
+        end
+      end
+
+      sort_desc(totals, opts) {|h, k, v| h.merge(k => v/similaritySums[k]) }
+    end
+
+    private
+
+    # Returns a hash containing the shared items between two different entities
+    #
+    # @param [Hash] Hash containing entity-item scores
+    # @param [String] Entity
+    # @param [String] Entity
+    #
+    # @return [Hash] Common items
+    def shared_items(scores, entity1, entity2)
+      Hash[*(scores[entity1].keys & scores[entity2].keys).flat_map{|k| [k, 1]}]
+    end
+
+    def sort_desc(results, opts={})
+      limit = opts[:limit] || 3
+
+      results.reduce({}) do |h, (k, v)|
+        yield(h, k, v)
+      end.sort_by{|k, v| v}.reverse[0..(opts[:limit] || 3)-1]
+    end
+  end
+
+  class PersonNotFound < StandardError; end
+end
+

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: pearson
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Alfonso Jiménez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.14.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.14.0
+description: Pearson correlation coefficient calculator
+email:
+- yo@alfonsojimenez.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- LICENSE.txt
+- .ruby-version
+- lib/pearson.rb
+homepage: https://github.com/alfonsojimenez/pearson
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Pearson correlation coefficient calculator
+test_files: []