rsvm 0.0.1

data/test/lib/cross_validation_test.rb

@@ -0,0 +1,35 @@
+require_relative '../test_helper'
+
+describe Svm::CrossValidation do
+  HEART_SCALE_CSV = File.join(File.dirname(__FILE__), '..', 'fixtures', 'heart_scale.csv')
+  UNBALANCED_CSV  = File.join(File.dirname(__FILE__), '..', 'fixtures', 'unbalanced.csv')
+
+  it "can perfom cross validation" do
+    problem = Svm::Problem.load_from_csv(HEART_SCALE_CSV)
+  
+    success_percentage = problem.results_for_cross_validation(3, :c => 10, :gamma => 0.01)
+    success_percentage.must_be :>, 0.8
+  end
+  
+  it "can find the best parameters for the problem" do 
+    problem = Svm::Problem.load_from_csv(HEART_SCALE_CSV)
+  
+    before = problem.results_for_cross_validation(3, :c => 1, :gamma => 0.001)
+  
+    options = problem.find_best_parameters
+  
+    after = problem.results_for_cross_validation(3, options)
+  
+    before.must_be :<, after
+  end
+
+  it "can cross validate unbalanced data" do
+    problem = Svm::Problem.load_from_csv(UNBALANCED_CSV)
+    
+    options = problem.find_best_parameters(3)
+    
+    success_percentage = problem.results_for_cross_validation(10, options)
+    
+    success_percentage.must_be :>, 0.8
+  end
+end

data/test/lib/model_test.rb

@@ -0,0 +1,84 @@
+require_relative '../test_helper'
+require 'tempfile'
+
+describe Svm::Model do
+
+  before do
+    @problem = Svm::Problem.new
+    @problem.data = [
+      [1, 1, 0, 1],
+      [-1, -1, 0, -1]
+    ]
+  end
+
+  it "stores the passed parameters" do
+    model = @problem.generate_model(:kernel_type => :linear, :c => 15)
+
+    param_struct = model.model_struct[:param]
+
+    param_struct[:kernel_type].must_equal(:linear)
+    param_struct[:c].must_equal(15)
+  end
+
+  it "can make a prediction for a sample" do
+    model = @problem.generate_model(:kernel_type => :linear, :c => 10)
+
+    model.predict([-1, 0, -1]).must_equal(-1)
+    model.predict([1,  0,  1]).must_equal(1)
+  end
+
+  it "must raise an exception if training parameters are not valid" do
+
+    proc {
+      @problem.generate_model(:kernel_type => :linear, :c => -10)
+    }.must_raise Svm::ParameterError
+
+  end
+
+  it "can be saved to a file" do
+    file = Tempfile.new('model')
+
+    begin
+       model = @problem.generate_model(:kernel_type => :linear, :c => 20)
+       model.save(file.path)
+
+       loaded_model = Svm::Model.load(file.path)
+
+       loaded_model.predict([-1, 0, -1]).must_equal(-1)
+       loaded_model.predict([1,  0,  1]).must_equal(1)
+    ensure
+       file.close
+       file.unlink   # deletes the temp file
+    end
+  end
+
+  it "stores the number of classes" do
+    model = @problem.generate_model(:kernel_type => :linear, :c => 20)
+    model.number_of_classes.must_equal 2
+  end
+
+  it "stores the labels" do
+    model = @problem.generate_model(:kernel_type => :linear, :c => 10)
+    model.labels.sort.must_equal [-1, 1]
+  end
+
+  it "can predict probabilities" do
+    csv_path = File.join(File.dirname(__FILE__), '..', 'fixtures', 'heart_scale.csv')
+    problem = Svm::Problem.load_from_csv(csv_path)
+
+    problem.estimate_probabilities = true
+
+    model = problem.generate_model
+
+    sample = [60.0, 1.0, 3.0, 140.0, 185.0, 0.0, 2.0, 155.0, 0.0, 3.0, 2.0, 0.0, 3.0]
+    probs = model.predict_probabilities(sample)
+
+    probs.values.each do |p|
+      p.must_be :>=, 0
+      p.must_be :<=, 1
+    end
+
+    probs.values.inject(&:+).must_be_close_to 1, 0.01
+  end
+
+end

data/test/lib/problem_test.rb

@@ -0,0 +1,39 @@
+require_relative '../test_helper'
+
+describe Svm::Problem do
+  
+  before do
+    @problem = Svm::Problem.new(:scale => false)
+    @problem.data = [
+      [1, 1, 0, 1],
+      [-1, -1, 0, -1],
+      [-1, -1, -1, -1]
+    ]
+  end
+  
+  it "stores the samples length" do
+    @problem.length.must_equal(3)
+  end
+    
+  it "stores problem values" do
+    @problem.value(0).must_equal(1.0)
+    @problem.value(1).must_equal(-1)
+  end
+  
+  it "stores samples features" do
+    @problem.sample(0).must_equal([1, 0, 1])
+    @problem.sample(1).must_equal([-1, 0, -1])
+  end
+  
+  it "can generate a model" do
+    model = @problem.generate_model
+    model.must_be_instance_of Svm::Model
+  end
+  
+  it "can suggets weights for labels" do
+    @problem.labels.sort.must_equal [-1, 1]
+    
+    @problem.suggested_labels_weights[1].must_equal(1.0/3)
+    @problem.suggested_labels_weights[-1].must_equal(2.0/3)
+  end
+end

data/test/lib/scaler_test.rb

@@ -0,0 +1,57 @@
+require_relative '../test_helper'
+
+describe Svm::Scaler do
+
+  before do
+    data = [
+      [1, 12.0, -7.6, 100_000, 0],
+      [2, 30.0, 0, -100_000, 0],
+      [3, 36.0, 7.6, 0, 0]
+    ]
+
+    @scaler = Svm::Scaler.new(data)
+    @scaler.scale!
+  end
+
+  it "doesn't scale the label data" do
+    @scaler.labels.must_equal([1, 2, 3])
+  end
+
+  it "finds the unit to scale" do
+    @scaler.unit[1].must_equal(12)
+    @scaler.unit[2].must_equal(7.6)
+    @scaler.unit[3].must_equal(100_000)
+    @scaler.unit[4].must_equal(0)
+  end
+
+  it "finds the zero to scale" do
+    @scaler.zero[1].must_equal(24.0)
+    @scaler.zero[2].must_equal(0)
+    @scaler.zero[3].must_equal(0)
+    @scaler.zero[4].must_equal(0)
+  end
+
+  it "scales the data" do
+    @scaler.values_for_feature(1).must_equal [-1, 0.5, 1]
+    @scaler.values_for_feature(2).must_equal [-1, 0, 1]
+    @scaler.values_for_feature(3).must_equal [1, -1, 0]
+    @scaler.values_for_feature(4).must_equal [0, 0, 0]
+  end
+  
+  it "can scale new input" do
+    @scaler.scale([36.0, -7.6, 10_000, 1]).must_equal [1, -1, 0.1, 1]
+  end
+
+  it "can be saved to a file" do
+    path = File.join(File.dirname(__FILE__), '..', 'fixtures', 'scaler.yml')
+
+    @scaler.save(path)
+
+    recovered = Svm::Scaler.load(path)
+    
+    recovered.scale([36.0, -7.6, 10_000, 1]).must_equal [1, -1, 0.1, 1]
+
+    File.delete(path)
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'minitest/spec'
+require 'minitest/autorun'
+require_relative '../lib/svm'

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: rsvm
+version: !ruby/object:Gem::Version
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors:
+- Alberto Fernández Capel
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: ffi
+  prerelease: false
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: FFI Ruby wrapper around libsvm
+email:
+- afcapel@gmail.com
+executables: []
+extensions:
+- ext/libsvm/extconf.rb
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- ext/libsvm/COPYRIGHT
+- ext/libsvm/FAQ.html
+- ext/libsvm/README
+- ext/libsvm/extconf.rb
+- ext/libsvm/svm.cpp
+- ext/libsvm/svm.h
+- lib/svm.rb
+- lib/svm/cross_validation.rb
+- lib/svm/debug.rb
+- lib/svm/model.rb
+- lib/svm/options.rb
+- lib/svm/problem.rb
+- lib/svm/scaler.rb
+- lib/svm/version.rb
+- rsvm.gemspec
+- test/fixtures/heart_scale.csv
+- test/fixtures/unbalanced.csv
+- test/lib/cross_validation_test.rb
+- test/lib/model_test.rb
+- test/lib/problem_test.rb
+- test/lib/scaler_test.rb
+- test/test_helper.rb
+homepage: https://github.com/afcapel/rsvm
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      segments:
+      - 0
+      hash: -3783083930879184987
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      segments:
+      - 0
+      hash: -3783083930879184987
+      version: '0'
+requirements: []
+rubyforge_project: rsvm
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Support Vector Machine Gem
+test_files:
+- test/fixtures/heart_scale.csv
+- test/fixtures/unbalanced.csv
+- test/lib/cross_validation_test.rb
+- test/lib/model_test.rb
+- test/lib/problem_test.rb
+- test/lib/scaler_test.rb
+- test/test_helper.rb