svmredlight 0.1.0

data/test/helper.rb

@@ -0,0 +1,19 @@
+require 'rubygems'
+require 'bundler'
+
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+require './lib/svmredlight'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+class Test::Unit::TestCase
+end

data/test/test_document.rb

@@ -0,0 +1,55 @@
+require './test/helper'
+include SVMLight
+
+class TestDocument < Test::Unit::TestCase
+
+  def test_create
+    d = Document.create(0, 0.5, 1, 0, [[1, 1.0 ], [4, 0.0] , [10, 0.0] ,[ 11, 0.5 ]])
+    assert_kind_of Document, d
+  end
+
+  def test_create_should_accept_integer_as_feature_weight
+    d = Document.create(0, 0.5, 1, 0, [[1, 0 ], [4, 0.0] , [10, 0.0] ,[ 11, 0.5 ]])
+    assert_kind_of Document, d
+  end
+
+  def test_create_using_new
+    d = Document.new({1 => 566.0, 4 => 133.0}, {docnum: 10, slackid: 1, queryid: 2, costfactor: 0.5}) 
+
+    assert_equal 10, d.docnum
+    assert_equal 1, d.slackid
+    assert_equal 2, d.queryid
+    assert_equal 0.5, d.costfactor
+  end
+
+  def test_should_be_able_to_access_properties
+    d1 = Document.create(0, 0.5, 1, 0, [[1, 1.0 ], [10, 0.0 ], [20, 0.0], [21, 0.1 ]])
+    d2 = Document.create(1, 0.6, 2, 1, [[1, 1.0 ], [30, 0.0 ], [40, 0.0], [41, 0.1 ]])
+
+    assert_equal 0, d1.docnum
+    assert_equal 1, d2.docnum
+
+    assert_equal 1, d1.slackid
+    assert_equal 2, d2.slackid
+
+    assert_equal 0, d1.queryid
+    assert_equal 1, d2.queryid
+
+    assert_equal 0.5, d1.costfactor
+    assert_equal 0.6, d2.costfactor
+  end
+
+  def test_all_word_numbers_should_be_greater_than_zero
+    assert_raise(ArgumentError){ Document.create(0, 0.5, 1, 0, [[0, 1.0 ], [10, 0.0 ], [20, 0.0], [21, 0.1 ]]) }
+    assert_raise(ArgumentError){ Document.create(1, 0.5, 1, 0, [[-1, 1.0 ], [30, 0.0 ], [40, 0.0], [41, 0.1 ]])}
+  end
+
+  def test_create_with_no_array
+    assert_raise(TypeError) { Document.create(-1, 0, 1, 0, {})  }
+  end
+
+  def test_create_with_empty_array
+    assert_raise(ArgumentError) { Document.create(-1, 0, 1, 0 [])}
+  end
+end
+

data/test/test_model.rb

@@ -0,0 +1,114 @@
+require './test/helper'
+include SVMLight
+
+class TestModel < Test::Unit::TestCase
+
+  def setup
+    @features ||= [
+      [ [1,0.6], [11, 0.0], [34, 0.1] ],
+      [ [5,0.4], [15, 0.0], [30, 0.1] ],
+      [ [1,0.1], [13, 0.0], [31, 0.1] ],
+      [ [7,0.7], [15, 0.0], [35, 0.1] ],
+      [ [5,0.6], [19, 0.0], [44, 0.1] ],
+    ]
+    @docs_and_labels ||= @features.each_with_index.map{|f,i| [ Document.create(i + 1, 1, 0, 0,  f), i%2 * -1]}
+  end
+
+  def test_learn_classification_with_alpha
+    m = Model.new(:classification, @docs_and_labels, {}, {}, [1, 0.0] * 50)
+    assert_kind_of Model, m
+
+    @docs_and_labels.each_with_index do |item, i|
+      assert_kind_of  Numeric, m.classify(item.first), "failed in item # #{i}"
+    end
+  end
+
+  def test_learn_classification
+    m = Model.new(:classification, @docs_and_labels, {}, {}, nil)
+    assert_kind_of Model, m
+    assert_equal 44, m.total_words
+    assert_equal 5, m.totdoc
+
+    @docs_and_labels.each_with_index do |item, i|
+      assert_kind_of  Numeric, m.classify(item.first), "failed in item # #{i}"
+    end
+
+  end
+
+  def test_learn_classification_with_learn_params
+    
+    learn_params = {
+       "predfile"            => "custom_file",
+       "alphafile"           => "alpha",
+       "biased_hyperplane"   => false,
+       "sharedslack"         => false,
+       "remove_inconsistent" => true
+    }
+
+    m = Model.new(:classification, @docs_and_labels, learn_params, {}, nil)
+    assert_kind_of Model, m
+
+    @docs_and_labels.each_with_index do |item, i|
+      assert_kind_of  Numeric, m.classify(item.first), "failed in item # #{i}"
+    end
+  end
+
+  def test_learn_classification_with_invalid_learn_params
+    learn_params = {"svm_c" =>  -1}
+    assert_raises(ArgumentError){Model.new(:classification, @docs_and_labels, learn_params, {}, nil)}
+    learn_params = {"svm_iter_to_shrink" =>  -1}
+    assert_raises(ArgumentError){Model.new(:classification, @docs_and_labels, learn_params, {}, nil)}
+  end
+
+  def test_learn_classification_with_kernel_params
+    
+    kernel_params = {
+      "poly_degree" => 3,
+      "rbf_gamma"   => 0.5,
+      "coef_lin"    => 0.4,
+      "coef_const"  => 0.56
+    }
+
+    m = Model.new(:classification, @docs_and_labels, {}, kernel_params, nil)
+    assert_kind_of Model, m
+
+    @docs_and_labels.each_with_index do |item, i|
+      assert_kind_of  Numeric, m.classify(item.first), "failed in item # #{i}"
+    end
+  end
+  
+  def test_learn_classification_with_learn_params_when_predfile_is_not_string
+    
+    learn_params = { "predfile"  => {}}
+  
+    assert_raise(ArgumentError) do
+      Model.new(:classification, @docs_and_labels, learn_params, {}, [1, 0.0, 1])
+    end
+
+  end
+
+  def test_learn_classification_fails_when_element_is_not_array
+    @docs_and_labels << []
+    assert_raises(ArgumentError){Model.new(:classification, @docs_and_labels, {}, {}, nil)}
+  end
+
+  def test_learn_classification_fails_when_element_is_arry_with_the_wrong_types
+    assert_raises(ArgumentError){Model.new(:classification, @docs_and_labels, {}, {}, [1, {}] )}
+  end
+
+  def test_read
+    assert m     = Model.read_from_file('test/assets/model')
+    assert_equal 3877,  m.support_vectors_count
+    assert_equal 39118, m.total_words
+  end
+
+  def test_classify
+    m = Model.read_from_file('test/assets/model')
+    assert_kind_of Numeric, m.classify( Document.create(-1, 1, 0, 0,[1.0, 0, 0, 0, 0.5 ].each_with_index.map{|v, i| [i + 1 ,v.to_f]} ) )
+    assert_kind_of Numeric, m.classify( Document.create(-1, 1, 0, 0,[1.0, 0, 0, 0, 0.5 ].each_with_index.map{|v, i| [i + 1,v.to_f]}) )
+    assert_kind_of Numeric, m.classify( Document.create(-1, 1, 0, 0,[1, 0, 0, 0, 0.8, 0, 0 , 0 ].each_with_index.map{|v, i| [i + 1,v.to_f]}) )
+    assert_kind_of Numeric, m.classify( Document.create(-1, 1, 0, 0,[1, 0.5, 0, 0, 0, 0 , 0 ].each_with_index.map{|v, i| [i + 1,v.to_f]}) )
+  end
+
+end
+

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification 
+name: svmredlight
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Camilo Lopez
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-09-11 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  prerelease: false
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: jeweler
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.6.0
+  type: :development
+  prerelease: false
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: ZenTest
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  prerelease: false
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: ruby-debug19
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  prerelease: false
+  version_requirements: *id005
+description: Ruby interface to SVMLight
+email: camilo@camilolopez.com
+executables: []
+
+extensions: 
+- ext/extconf.rb
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- .document
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- VERSION
+- examples/example1/example.rb
+- examples/example1/test.dat
+- examples/example1/train.dat
+- examples/example1/words
+- ext/extconf.rb
+- ext/svmredlight.c
+- lib/svmredlight.rb
+- lib/svmredlight/document.rb
+- lib/svmredlight/model.rb
+- svmredlight.gemspec
+- test/assets/model
+- test/helper.rb
+- test/test_document.rb
+- test/test_model.rb
+has_rdoc: true
+homepage: http://github.com/camilo/svmredlight
+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: 2455726499843414946
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.0
+signing_key: 
+specification_version: 3
+summary: Ruby interface to SVMLight
+test_files: []
+