coletivo 0.0.0

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Diógenes Falcão
+
+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/README.rdoc

@@ -0,0 +1,44 @@
+= coletivo
+
+An awesome, flexible, powerful, useful, tricky and liar Rails 3 recommendations engine.
+
+== Installation:
+
+  sudo gem install coletivo
+  rails g coletivo
+  rake db:migrate
+
+== Usage:
+
+At your Rails model that represents a person (can be an _User_, _Member_, or something like that):
+
+  class User < ActiveRecord::Base
+    has_own_preferences
+
+    # ...
+  end
+
+  # ratings
+  current_user = User.create(:name => 'Diogenes')
+  movie = Movie.create(:name => 'The Tourist', :year => 2010)
+
+  current_user.rate!(movie, 4.5)
+
+  # after a lot of ratings... recommendations.
+  Movie.find_recommendations_for(current_user) # => movies and more movies...
+
+== Contributing to coletivo
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
+* Fork the project
+* Start a feature/bugfix branch
+* Commit and push until you are happy with your contribution
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+== Copyright
+
+Copyright (c) 2011 Diógenes Falcão. See LICENSE.txt for
+further details.
+

data/lib/coletivo.rb

@@ -0,0 +1,35 @@
+require 'rails'
+require 'active_model'
+require 'active_record'
+require 'active_support'
+
+module Coletivo
+  module Models
+    autoload :Recommendable, 'coletivo/models/recommendable'
+    autoload :Person, 'coletivo/models/person'
+    autoload :PersonRating, 'coletivo/models/person_rating'
+  end
+
+  module Similarity
+    NO_SIMILARITY = -1.0..0.49
+    SIMILAR = 0.5..0.99
+    IDENTICAL = 1.0
+
+    autoload :BaseStrategy, 'coletivo/similarity/base_strategy'
+    autoload :EuclideanDistanceStrategy, 'coletivo/similarity/euclidean_distance_strategy'
+    autoload :PearsonCorrelationStrategy, 'coletivo/similarity/pearson_correlation_strategy'
+    autoload :Engine, 'coletivo/similarity/engine'
+  end
+
+  module Config
+    mattr_accessor :ratings_container
+
+    # Defaults
+    self.ratings_container = Coletivo::Models::PersonRating
+  end
+
+  if defined?(Rails)
+    require 'coletivo/rails/engine'
+    require 'coletivo/rails/active_record'
+  end
+end

data/lib/coletivo/models/person.rb

@@ -0,0 +1,27 @@
+module Coletivo
+  module Models
+    module Person
+      def self.included(base)
+        base.extend ClassMethods
+      end
+
+      module ClassMethods
+        # TODO: has_own_preferences doc.
+        def has_own_preferences(options = {})
+          self.send :include, InstanceMethods
+        end
+      end # ClassMethods
+
+      module InstanceMethods
+        def rate!(rateable, weight)
+          Coletivo::Config.ratings_container.create!({
+            :person => self,
+            :rateable => rateable,
+            :weight => weight
+          })
+        end
+      end # InstanceMethods
+
+    end # Person
+  end # Models
+end

data/lib/coletivo/models/person_rating.rb

@@ -0,0 +1,14 @@
+module Coletivo
+  module Models
+    class PersonRating < ActiveRecord::Base
+      belongs_to :person, :polymorphic => true
+      belongs_to :rateable, :polymorphic => true
+
+      validates :person, :rateable, :weight, :presence => true
+
+      def self.find_for_recommendation(recommendable)
+        find(:all)
+      end
+    end
+  end
+end

data/lib/coletivo/models/recommendable.rb

@@ -0,0 +1,96 @@
+module Coletivo
+  module Models
+    module Recommendable
+      def self.included(base)
+        base.extend ClassMethods
+        base.send :include, InstanceMethods
+      end
+
+      module ClassMethods
+        def find_recommendations_for(model, options = {})
+          sim_totals, weighted_means = {}, {}
+          preferences = options[:preferences] ||
+            load_preferences_for_recommendation(model)
+
+          preferences.each do |other, other_prefs|
+            next if other == model
+
+            sim = model.similarity_with(other, options)
+            next if sim <= 0
+
+            other_prefs.each do |item, weight|
+              unless preferences[model.id].keys.include?(item)
+                sim_totals[item] ||= 0
+                weighted_means[item] ||= 0
+
+                sim_totals[item] += sim
+                weighted_means[item] += weight * sim
+              end
+            end
+          end
+
+          # DESC sort by weighted mean of ratings
+          # e.g: [[5.35, "movie_2"], [2.0, "movie_4"]]
+          top = weighted_means.collect { |i, mean| [mean/sim_totals[i], i] }\
+            .sort { |t_one, t_other| t_other <=> t_one }
+
+          ids = top.collect(&:last) # e.g: ['movie_2', 'movie_4']
+          models = where(:id => ids)
+
+          top.collect { |weight, item| models.detect {|m| m.id == item } }\
+            .compact
+        end
+
+        def find_ratings_for_recommendation(recommendable)
+          Coletivo::Config.ratings_container\
+            .find_for_recommendation(recommendable)
+        end
+
+        # Map a preferences Hash by a collection of ratings.
+        #
+        # Rating objects have a +person+, a +rateable+ object and a +weight+.
+        #
+        # The preferences can be mapped for _persons_, like that:
+        #   {
+        #      :person_1 => {:movie_1 => 2.0, :movie_2 => 5.0},
+        #      :person_2 => {:movie_1 => 3.0, :movie_2 => 4.0}
+        #   }
+        #
+        # Or for _rateable_ items, like that:
+        #   {
+        #      :movie_1 => {:person_1 => 2.0, :person_2 => 3.0},
+        #      :movie_2 => {:person_1 => 5.0, :person_2 => 4.0}
+        #   }
+        #
+        # Expected keys are :person or :rateable
+        def map_ratings_to_preferences(ratings)
+          #TODO: ?? #preferences_by_ratings(:rateable, :person): Item based rec.
+          key, subkey = :person, :rateable
+          preferences = {}
+
+          ratings.each do |rating|
+            p = preferences[rating.send(key).id] ||= {}
+            p[rating.send(subkey).id] = rating.weight
+          end
+
+          preferences
+        end
+
+        def load_preferences_for_recommendation(model)
+          r = model.class.find_ratings_for_recommendation(model)
+          model.class.map_ratings_to_preferences(r)
+        end
+      end
+
+      module InstanceMethods
+        def similarity_with(other_id, options = {})
+          p = options[:preferences] ||
+            self.class.load_preferences_for_recommendation(self)
+
+          Coletivo::Similarity::Engine\
+            .similarity_between(self.id, other_id, p, options)
+        end
+      end
+    end # Recommendable
+  end # Models
+end

data/lib/coletivo/rails/active_record.rb

@@ -0,0 +1,2 @@
+ActiveRecord::Base.send :include, Coletivo::Models::Person
+ActiveRecord::Base.send :include, Coletivo::Models::Recommendable

data/lib/coletivo/rails/engine.rb

@@ -0,0 +1,6 @@
+require 'coletivo'
+
+module Coletivo
+  class Engine < Rails::Engine
+  end
+end

data/lib/coletivo/similarity/base_strategy.rb

@@ -0,0 +1,25 @@
+module Coletivo
+  module Similarity
+    class BaseStrategy
+      attr_accessor :preferences
+
+      def similarity_between(one, other)
+        raise "The #similarity_between was not implemented in #{self.class}"
+      end
+
+      def train_with(people_preferences)
+        @preferences = people_preferences
+      end
+
+      protected
+
+      def shared_items_between(one, other)
+        return [] unless preferences[one] && preferences[other]
+
+        preferences[one].keys.select { |item|
+          preferences[other].keys.include? item
+        }
+      end
+    end
+  end # Similarity
+end # Coletivo

data/lib/coletivo/similarity/engine.rb

@@ -0,0 +1,22 @@
+module Coletivo
+  module Similarity
+    class Engine
+      def self.similarity_between(one, other, preferences, options = {})
+        strategy = load_strategy options[:strategy]
+        strategy.train_with(preferences)
+
+        strategy.similarity_between(one, other)
+      end
+
+      protected
+
+      def self.load_strategy(key)
+        if :pearson == key
+          Coletivo::Similarity::PearsonCorrelationStrategy.new
+        else
+          Coletivo::Similarity::EuclideanDistanceStrategy.new
+        end
+      end
+    end # Engine
+  end # Similarity
+end

data/lib/coletivo/similarity/euclidean_distance_strategy.rb

@@ -0,0 +1,18 @@
+module Coletivo
+  module Similarity
+    class EuclideanDistanceStrategy < BaseStrategy
+      def similarity_between(one, other)
+        shared = shared_items_between(one, other)
+
+        return 0 if shared.empty?
+
+        sum_of_squares = shared.inject(0.0) { |sum, item|
+          sum + (preferences[one][item] - preferences[other][item]) ** 2
+        }
+
+        1 / (1 + sum_of_squares)
+      end
+    end
+  end
+end # Similarity
+

data/lib/coletivo/similarity/pearson_correlation_strategy.rb

@@ -0,0 +1,35 @@
+module Coletivo
+  module Similarity
+    class PearsonCorrelationStrategy < BaseStrategy
+      def similarity_between(one, other)
+        shared      = shared_items_between(one, other)
+        prefs_one   = preferences[one]
+        prefs_other = preferences[other]
+
+        return 0 if shared.empty?
+
+        sum_prefs_one = sum_prefs_other = sum_squares_one = \
+          sum_squares_other = p_sum = 0.0
+
+        shared.each { |item|
+          sum_prefs_one     += prefs_one[item]
+          sum_prefs_other   += prefs_other[item]
+          sum_squares_one   += prefs_one[item] ** 2
+          sum_squares_other += prefs_other[item] ** 2
+          p_sum             += prefs_one[item] * prefs_other[item]
+        }
+
+        total_shared = shared.size
+
+        numerator = p_sum - (sum_prefs_one * sum_prefs_other / total_shared)
+
+        den_one = sum_squares_one - (sum_prefs_one ** 2) / total_shared
+        den_other = sum_squares_other - (sum_prefs_other ** 2) / total_shared
+
+        denominator = Math.sqrt(den_one * den_other)
+
+        denominator == 0 ? 0 : numerator / denominator
+      end
+    end
+  end
+end

data/lib/generators/coletivo/coletivo_generator.rb

@@ -0,0 +1,24 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+
+class ColetivoGenerator < Rails::Generators::Base
+  include Rails::Generators::Migration
+
+  def self.source_root
+    File.join(File.dirname(__FILE__), 'templates')
+  end
+
+  # Implement the required interface for Rails::Generators::Migration.
+  def self.next_migration_number(dirname) #:nodoc:
+    if ActiveRecord::Base.timestamped_migrations
+      Time.now.utc.strftime("%Y%m%d%H%M%S")
+    else
+      "%.3d" % (current_migration_number(dirname) + 1)
+    end
+  end
+
+  def create_migration_file
+    migration_template 'person_ratings_migration.rb',
+      'db/migrate/create_person_ratings.rb'
+  end
+end

data/lib/generators/coletivo/templates/person_ratings_migration.rb

@@ -0,0 +1,19 @@
+class CreatePersonRatings < ActiveRecord::Migration
+  def self.up
+    create_table :person_ratings do |t|
+      t.integer :person_id
+      t.string  :person_type
+
+      t.integer :rateable_id
+      t.string  :rateable_type
+
+      t.decimal :weight, :precision => 5, :scale => 2
+
+      t.timestamps
+    end
+  end
+
+  def self.down
+    drop_table :person_ratings
+  end
+end

metadata

@@ -0,0 +1,159 @@
+--- !ruby/object:Gem::Specification 
+name: coletivo
+version: !ruby/object:Gem::Version 
+  hash: 31
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 0
+  version: 0.0.0
+platform: ruby
+authors: 
+- "Di\xC3\xB3genes Falc\xC3\xA3o"
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-08 00:00:00 -03:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  name: rails
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 3
+        - 0
+        - 7
+        version: 3.0.7
+  requirement: *id001
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  name: shoulda
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: -1848230022
+        segments: 
+        - 3
+        - 0
+        - 0
+        - beta2
+        version: 3.0.0.beta2
+  requirement: *id002
+  type: :development
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  name: bundler
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 11
+        segments: 
+        - 1
+        - 0
+        - 14
+        version: 1.0.14
+  requirement: *id003
+  type: :development
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  name: jeweler
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 11
+        segments: 
+        - 1
+        - 6
+        - 2
+        version: 1.6.2
+  requirement: *id004
+  type: :development
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  name: turn
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id005
+  type: :development
+description: An awesome, flexible, powerful, useful, tricky and liar Rails 3 recommendations engine.
+email: diogenes.araujo@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- lib/coletivo.rb
+- lib/coletivo/models/person.rb
+- lib/coletivo/models/person_rating.rb
+- lib/coletivo/models/recommendable.rb
+- lib/coletivo/rails/active_record.rb
+- lib/coletivo/rails/engine.rb
+- lib/coletivo/similarity/base_strategy.rb
+- lib/coletivo/similarity/engine.rb
+- lib/coletivo/similarity/euclidean_distance_strategy.rb
+- lib/coletivo/similarity/pearson_correlation_strategy.rb
+- lib/generators/coletivo/coletivo_generator.rb
+- lib/generators/coletivo/templates/person_ratings_migration.rb
+- LICENSE.txt
+- README.rdoc
+has_rdoc: true
+homepage: http://github.com/diogenes/coletivo
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: An awesome, flexible, powerful, useful, tricky and liar Rails 3 recommendations engine.
+test_files: []
+