jruby_mahout 0.2.0

data/spec/recommender_spec.rb

@@ -0,0 +1,296 @@
+require 'spec_helper'
+
+describe JrubyMahout::Recommender do
+  describe ".new" do
+    context "with valid arguments" do
+      it "should return an instance of JrubyMahout::Recommender for PearsonCorrelationSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for EuclideanDistanceSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for SpearmanCorrelationSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for LogLikelihoodSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("LogLikelihoodSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for TanimotoCoefficientSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for GenericItemSimilarity and GenericUserBasedRecommender" do
+        JrubyMahout::Recommender.new("GenericItemSimilarity", 5, "GenericUserBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for PearsonCorrelationSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for EuclideanDistanceSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for SpearmanCorrelationSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for LogLikelihoodSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("LogLikelihoodSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for TanimotoCoefficientSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for GenericItemSimilarity and GenericItemBasedRecommender" do
+        JrubyMahout::Recommender.new("GenericItemSimilarity", nil, "GenericItemBasedRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+
+      it "should return an instance of JrubyMahout::Recommender for SlopeOneRecommender" do
+        JrubyMahout::Recommender.new(nil, nil, "SlopeOneRecommender", false).should
+        be_an_instance_of JrubyMahout::Recommender
+      end
+    end
+  end
+
+  describe "data_model=" do
+    it "should load file data model" do
+      recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false)
+      recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+      recommender.data_model.should be_an_instance_of org.apache.mahout.cf.taste.impl.model.file.FileDataModel
+    end
+
+    it "should load postgres data model" do
+      recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false)
+      recommender.data_model = JrubyMahout::DataModel.new("postgres", {
+          :host => "localhost",
+          :port => 5432,
+          :db_name => "postgres",
+          :username => "postgres",
+          :password => "postgres",
+          :table_name => "taste_preferences"
+      }).data_model
+
+      recommender.data_model.should be_an_instance_of org.apache.mahout.cf.taste.impl.model.jdbc.PostgreSQLJDBCDataModel
+    end
+  end
+
+  describe ".recommend" do
+    context "with valid arguments" do
+      it "should return an array for PearsonCorrelationSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for EuclideanDistanceSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for SpearmanCorrelationSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for LogLikelihoodSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("LogLikelihoodSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for TanimotoCoefficientSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for GenericItemSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("GenericItemSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for PearsonCorrelationSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for EuclideanDistanceSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for LogLikelihoodSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("LogLikelihoodSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for TanimotoCoefficientSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for GenericItemSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("GenericItemSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+
+      it "should return an array for SlopeOneRecommender" do
+        recommender = JrubyMahout::Recommender.new("nil", nil, "SlopeOneRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be_an_instance_of Array
+      end
+    end
+
+    context "with invalid arguments" do
+      it "should return nil for SpearmanCorrelationSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.recommend(1, 10, nil).should be nil
+      end
+    end
+  end
+
+  describe ".evaluate" do
+    context "with valid arguments" do
+      it "should return a float" do
+        recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return an array for PearsonCorrelationSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for EuclideanDistanceSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for SpearmanCorrelationSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for LogLikelihoodSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("LogLikelihoodSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for TanimotoCoefficientSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for GenericItemSimilarity and GenericUserBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("GenericItemSimilarity", 5, "GenericUserBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for PearsonCorrelationSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for EuclideanDistanceSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("EuclideanDistanceSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for LogLikelihoodSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("LogLikelihoodSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for TanimotoCoefficientSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("TanimotoCoefficientSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for GenericItemSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("GenericItemSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+
+      it "should return a float for SlopeOneRecommender" do
+        recommender = JrubyMahout::Recommender.new("nil", nil, "SlopeOneRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be_an_instance_of Float
+      end
+    end
+
+    context "with invalid arguments" do
+      it "should return nil for SpearmanCorrelationSimilarity and GenericItemBasedRecommender" do
+        recommender = JrubyMahout::Recommender.new("SpearmanCorrelationSimilarity", nil, "GenericItemBasedRecommender", false)
+        recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "spec/recommender_data.csv" }).data_model
+
+        recommender.evaluate(0.7, 0.3).should be nil
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'jruby_mahout'
+
+RSpec.configure do |config|
+
+end

metadata

@@ -0,0 +1,110 @@
+--- !ruby/object:Gem::Specification
+name: jruby_mahout
+version: !ruby/object:Gem::Version
+  prerelease:
+  version: 0.2.0
+platform: ruby
+authors:
+- Vasily Vasinov
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2012-12-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: !binary |-
+          MA==
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: !binary |-
+          MA==
+    none: false
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: rspec
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: !binary |-
+          MA==
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: !binary |-
+          MA==
+    none: false
+  prerelease: false
+  type: :development
+description: Jruby Mahout is a gem that unleashes the power of Apache Mahout in the world of Jruby. Mahout is a superior machine learning library written in Java. It deals with recommendations, clustering and classification machine learning problems at scale. Until now it was difficult to use it in Ruby projects. You'd have to implement Java interfaces in Jruby yourself, which is not quick especially if you just started exploring the world of machine learning.
+email:
+- vasinov@me.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- Gemfile.lock
+- MIT-LICENSE
+- README.md
+- Rakefile
+- jruby_mahout.gemspec
+- lib/jruby_mahout.rb
+- lib/jruby_mahout/data_model.rb
+- lib/jruby_mahout/evaluator.rb
+- lib/jruby_mahout/mahout_imports.rb
+- lib/jruby_mahout/mysql_manager.rb
+- lib/jruby_mahout/postgres_manager.rb
+- lib/jruby_mahout/recommender.rb
+- lib/jruby_mahout/recommender_builder.rb
+- lib/jruby_mahout/version.rb
+- spec/recommender_data.csv
+- spec/recommender_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/vasinov/jruby_mahout
+licenses:
+- MIT
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: !binary |-
+        MA==
+  none: false
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: !binary |-
+        MA==
+  none: false
+requirements: []
+rubyforge_project:
+rubygems_version: 1.8.24
+signing_key:
+specification_version: 3
+summary: Jruby Mahout is a gem that unleashes the power of Apache Mahout in the world of Jruby.
+test_files:
+- !binary |-
+  c3BlYy9yZWNvbW1lbmRlcl9kYXRhLmNzdg==
+- !binary |-
+  c3BlYy9yZWNvbW1lbmRlcl9zcGVjLnJi
+- !binary |-
+  c3BlYy9zcGVjX2hlbHBlci5yYg==