jruby_mahout 0.2.1 → 0.2.2

data/README.md

@@ -1,30 +1,76 @@
-# Jruby Mahout
-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.
+# JRuby Mahout
+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.
 
-The goal of this library is to make machine learning at scale in Jruby projects simple.
+The goal of this library is to make machine learning at scale in JRuby projects simple.
 
 ## Quick Overview
-This is an early version of a Jruby gem that only supports Mahout recommendations. It also includes a simple Postgres manager that can be used to manage appropriate recommendations tables. Unfortunately it's impossible to use ActiveRecord (AR) with Mahout, because AR at a mach higher level and creates a lot of overhead that is critical when dealing with millions of records in real time.
+This is an early version of a JRuby gem that only supports Mahout recommendations. It also includes a simple Postgres manager that can be used to manage appropriate recommendations tables. Unfortunately it's impossible to use ActiveRecord (AR) with Mahout, because AR operates at a much higher level and creates a lot of overhead that is critical when dealing with millions of records in real time.
 
 ## Get Mahout
-First of all you need to download Mahout library from one of the [mirrors](http://www.apache.org/dyn/closer.cgi/mahout/). Jruby Mahout only supports Mahout 0.7 at this point.
+First of all you need to download the Mahout library from one of the [mirrors](http://www.apache.org/dyn/closer.cgi/mahout/). Jruby Mahout only supports Mahout 0.7 at this point.
 
 ## Get Postgres JDBC Adapter
-If you wish to work with a database for recommendations, you'll have to install [JDBC driver for Postgres](http://jdbc.postgresql.org/download.html). Another option is to use file-based recommendation.
+If you wish to work with a database for recommendations, you'll have to install the [JDBC driver for Postgres](http://jdbc.postgresql.org/download.html). Another option is to use file-based recommendations.
 
 ## Installation
-### 1. Set environment variable MAHOUT_DIR to point at your Mahout installation.
+### 1. Set the environment variable MAHOUT_DIR to point at your Mahout installation.
 ### 2. Add the gem to your `Gemfile`
 ```ruby
 platform :jruby do
   gem "jruby_mahout"
 end
 ```
-And run `bundle install`.
+### 3. Run `bundle install`.
+
+## How to Use?
+I am planning to add more examples covering Jruby Mahout use cases to [this repo](https://github.com/vasinov/jruby_mahout-examples) soon.
+
+First, define the `MAHOUT_DIR` environmental variable for your Mahout installation. For example:
+
+```
+export MAHOUT_DIR=/bin/mahout
+```
+
+The easiest way to start working with Jruby Mahout recommendations is to initialize a recommender:
+```ruby
+require 'jruby_mahout'
+recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", 5, "GenericUserBasedRecommender", false)
+```
+
+Set up a data model:
+```ruby
+recommender.data_model = JrubyMahout::DataModel.new("file", { :file_path => "recommender_data.csv" }).data_model
+```
+
+and get recommendations:
+```ruby
+puts recommender.recommend(2, 10, nil) # 10 recommendations for user with id = 2
+```
+
+You can evaluate your recommender to see how efficient it is:
+```ruby
+puts recommender.evaluate(0.7, 0.3)
+```
+
+The closer the score is to zero—the better.
+
+I realize that it's a very sparse introduction to Jruby Mahout. I am working on the tutorial and better documentation that should cover this gem more in depth. Stay tuned at [my blog](http://www.vasinov.com/blog).
+
+## Development Plans
+There are several things that should be supported by this gem, before it can be used in production. Some of them are:
+- Hadoop integration
+- Clustering support
+- Classification support
+- Better docs
+
+If you feel like you can help—please do.
+
+## Testing
+Jruby Mahout is thoroughly tested with Rspec.
 
 ## Contribute
 - Fork the project.
 - Write code for a feature or bug fix.
 - Add Rspec tests for it.
 - Commit, do not make changes to rakefile or version.
-- Submit a pull request.
+- Submit a pull request.

data/lib/jruby_mahout/postgres_manager.rb

@@ -44,19 +44,27 @@ module JrubyMahout
       end
     end
 
-    def upsert_record(record, name)
+    def upsert_record(table_name, record)
       begin
-        @statement.execute("UPDATE #{name} SET user_id=#{record[:user_id]}, item_id=#{record[:item_id]}, rating=#{record[:rating]} WHERE user_id=#{record[:user_id]} AND item_id=#{record[:item_id]};")
-        @statement.execute("INSERT INTO #{name} (user_id, item_id, rating) SELECT #{record[:user_id]}, #{record[:item_id]}, #{record[:rating]} WHERE NOT EXISTS (SELECT 1 FROM #{name} WHERE user_id=#{record[:user_id]} AND item_id=#{record[:item_id]});")
+        @statement.execute("UPDATE #{table_name} SET user_id=#{record[:user_id]}, item_id=#{record[:item_id]}, rating=#{record[:rating]} WHERE user_id=#{record[:user_id]} AND item_id=#{record[:item_id]};")
+        @statement.execute("INSERT INTO #{table_name} (user_id, item_id, rating) SELECT #{record[:user_id]}, #{record[:item_id]}, #{record[:rating]} WHERE NOT EXISTS (SELECT 1 FROM #{table_name} WHERE user_id=#{record[:user_id]} AND item_id=#{record[:item_id]});")
       rescue java.sql.SQLException => e
         puts e
       end
     end
 
-    def create_table(name)
+    def delete_record(table_name, record)
+      begin
+        @statement.execute("DELETE FROM #{table_name} WHERE user_id=#{record[:user_id]} AND item_id=#{record[:item_id]};")
+      rescue java.sql.SQLException => e
+        puts e
+      end
+    end
+
+    def create_table(table_name)
       begin
         @statement.executeUpdate("
-          CREATE TABLE #{name} (
+          CREATE TABLE #{table_name} (
             user_id BIGINT NOT NULL,
             item_id BIGINT NOT NULL,
             rating int NOT NULL,
@@ -64,18 +72,18 @@ module JrubyMahout
             PRIMARY KEY (user_id, item_id)
           );
         ")
-        @statement.executeUpdate("CREATE INDEX #{name}_user_id_index ON #{name} (user_id);")
-        @statement.executeUpdate("CREATE INDEX #{name}_item_id_index ON #{name} (item_id);")
+        @statement.executeUpdate("CREATE INDEX #{table_name}_user_id_index ON #{table_name} (user_id);")
+        @statement.executeUpdate("CREATE INDEX #{table_name}_item_id_index ON #{table_name} (item_id);")
       rescue java.sql.SQLException => e
         puts e
       end
     end
 
-    def delete_table(name)
+    def delete_table(table_name)
       begin
-        @statement.executeUpdate("DROP INDEX IF EXISTS #{name}_user_id_index;")
-        @statement.executeUpdate("DROP INDEX IF EXISTS #{name}_item_id_index;")
-        @statement.executeUpdate("DROP TABLE IF EXISTS #{name};")
+        @statement.executeUpdate("DROP INDEX IF EXISTS #{table_name}_user_id_index;")
+        @statement.executeUpdate("DROP INDEX IF EXISTS #{table_name}_item_id_index;")
+        @statement.executeUpdate("DROP TABLE IF EXISTS #{table_name};")
       rescue java.sql.SQLException => e
         puts e
       end

data/lib/jruby_mahout/recommender.rb

@@ -34,18 +34,18 @@ module JrubyMahout
     end
 
     def similar_items(item_id, number_of_items, rescorer)
-      if @recommender.nil? or @recommender_name == "GenericItemBasedRecommender"
+      if @recommender.nil? or @recommender_name == "GenericUserBasedRecommender"
         nil
       else
-        @recommender.mostSimilarItems(item_id, number_of_items, rescorer)
+        to_array(@recommender.mostSimilarItems(item_id, number_of_items, rescorer))
       end
     end
 
-    def similar_users(user_id, number_of_items, rescorer)
-      if @recommender.nil? or @recommender_name == "GenericUserBasedRecommender"
+    def similar_users(user_id, number_of_users, rescorer)
+      if @recommender.nil? or @recommender_name == "GenericItemBasedRecommender"
         nil
       else
-        @recommender.mostSimilarUserIDs(user_id, amount, rescorer)
+        to_array(@recommender.mostSimilarUserIDs(user_id, number_of_users, rescorer))
       end
     end
 
@@ -58,10 +58,10 @@ module JrubyMahout
     end
 
     def recommended_because(user_id, item_id, number_of_items)
-      if @recommender.nil? or @recommender_name == "GenericItemBasedRecommender"
+      if @recommender.nil? or @recommender_name == "GenericUserBasedRecommender"
         nil
       else
-        @recommender.recommendedBecause(user_id, item_id, number_of_items)
+        to_array(@recommender.recommendedBecause(user_id, item_id, number_of_items))
       end
     end
 
@@ -74,5 +74,15 @@ module JrubyMahout
 
       recommendations_array
     end
+
+    private
+    def to_array(things)
+      things_array = []
+      things.each do |thing_id|
+        things_array << thing_id
+      end
+
+      things_array
+    end
   end
 end

data/lib/jruby_mahout/recommender_builder.rb

@@ -9,6 +9,7 @@ module JrubyMahout
     java_import org.apache.mahout.cf.taste.impl.similarity.PearsonCorrelationSimilarity
 
     java_import org.apache.mahout.cf.taste.impl.neighborhood.NearestNUserNeighborhood
+    java_import org.apache.mahout.cf.taste.impl.neighborhood.ThresholdUserNeighborhood
 
     java_import org.apache.mahout.cf.taste.impl.recommender.GenericUserBasedRecommender
     java_import org.apache.mahout.cf.taste.impl.recommender.GenericItemBasedRecommender
@@ -49,7 +50,11 @@ module JrubyMahout
         end
 
         unless @neighborhood_size.nil?
-          neighborhood = NearestNUserNeighborhood.new(Integer(@neighborhood_size), similarity, data_model)
+          if @neighborhood_size > 1
+            neighborhood = NearestNUserNeighborhood.new(Integer(@neighborhood_size), similarity, data_model)
+          elsif @neighborhood_size >= -1 and @neighborhood_size <= 1
+            neighborhood = ThresholdUserNeighborhood.new(Float(@neighborhood_size), similarity, data_model)
+          end
         end
 
         case @recommender_name

data/lib/jruby_mahout/version.rb

@@ -1,3 +1,3 @@
 module JrubyMahout
-  VERSION = '0.2.1'
+  VERSION = '0.2.2'
 end

data/spec/recommender_spec.rb

@@ -95,88 +95,176 @@ describe JrubyMahout::Recommender do
 
   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
+      context "with NearestNUserNeighborhood" 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 ThresholdUserNeighborhood" do
+        it "should return an array for PearsonCorrelationSimilarity and GenericUserBasedRecommender" do
+          recommender = JrubyMahout::Recommender.new("PearsonCorrelationSimilarity", 0.7, "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", 0.7, "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", 0.7, "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", 0.7, "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", 0.7, "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", 0.7, "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
 
-      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
+          recommender.recommend(1, 10, nil).should be_an_instance_of Array
+        end
       end
     end
 
@@ -293,4 +381,52 @@ describe JrubyMahout::Recommender do
       end
     end
   end
+
+  # TODO: cover all cases
+  describe ".similar_users" do
+    context "with valid arguments" do
+      it "should return an array of users" 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.similar_users(1, 10, nil).should be_an_instance_of Array
+      end
+    end
+  end
+
+  # TODO: cover all cases
+  describe ".similar_items" do
+    context "with valid arguments" do
+      it "should return an array of items" 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.similar_items(4, 10, nil).should be_an_instance_of Array
+      end
+    end
+  end
+
+  # TODO: cover all cases
+  describe ".recommended_because" do
+    context "with valid arguments" do
+      it "should return an array of items" 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.recommended_because(1, 138, 5).should be_an_instance_of Array
+      end
+    end
+  end
+
+  # TODO: cover all cases
+  describe ".estimate_preference" do
+    context "with valid arguments" do
+      it "should return afloat with an estimate" 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.estimate_preference(1, 138).should be_an_instance_of Float
+      end
+    end
+  end
 end

metadata

@@ -2,14 +2,14 @@
 name: jruby_mahout
 version: !ruby/object:Gem::Version
   prerelease:
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Vasily Vasinov
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2012-12-10 00:00:00.000000000 Z
+date: 2012-12-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -47,7 +47,7 @@ dependencies:
     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.
+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: []
@@ -60,8 +60,6 @@ files:
   bGliL2pydWJ5X21haG91dC9kYXRhX21vZGVsLnJi
 - !binary |-
   bGliL2pydWJ5X21haG91dC9ldmFsdWF0b3IucmI=
-- !binary |-
-  bGliL2pydWJ5X21haG91dC9tYWhvdXRfaW1wb3J0cy5yYg==
 - !binary |-
   bGliL2pydWJ5X21haG91dC9teXNxbF9tYW5hZ2VyLnJi
 - !binary |-
@@ -107,7 +105,7 @@ 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.
+summary: JRuby Mahout is a gem that unleashes the power of Apache Mahout in the world of JRuby.
 test_files:
 - !binary |-
   c3BlYy9yZWNvbW1lbmRlcl9kYXRhLmNzdg==

data/lib/jruby_mahout/mahout_imports.rb

@@ -1,21 +0,0 @@
-# Recommenders
-
-
-# Neighborhoods
-
-
-# Recommenders
-
-
-# Weighting
-
-
-# Evaluators
-
-
-# Data Models
-
-
-
-# Postgres
-