ruby_linear 0.1.0

data/spec/integration_spec.rb

@@ -0,0 +1,54 @@
+require 'spec_helper'
+
+
+describe(RubyLinear) do
+  
+  # The data files are the data set from http://www.csie.ntu.edu.tw/~cjlin/libsvmtools/datasets/multiclass.html#dna
+  # The model was created with train -s 5 -e 0.01 -B 1 dna.scale.txt dna.dat
+  # the output was created with predict dna.scale.t dna.dat dna.out
+  
+  def parse_line(l)
+    h = {}
+    l.split(' ').each do |s|
+      if s =~ /(\d+):(\d+)/
+        h[$1.to_i] = $2.to_i
+      end
+    end
+    h
+  end
+  
+  before(:each) do
+    problem = RubyLinear::Problem.load_file("spec/fixtures/dna.scale.txt",1)
+    @model = RubyLinear::Model.new(problem, :solver => RubyLinear::L1R_L2LOSS_SVC)
+  end
+  
+  it 'should produce the same output as the command line tools' do    
+    @model.should predict_values('spec/fixtures/dna.out').for_input('spec/fixtures/dna.scale.t')
+  end
+  
+  RSpec::Matchers.define :predict_values do |output_file|
+    match do |model|
+      input_lines = File.readlines(@input)
+      output_lines = File.readlines(output_file)
+      raise "mismatched inputs" unless input_lines.length == output_lines.length
+      raise "input empty" if input_lines.length < 1
+      input_lines.each_with_index do |line, index|
+        sample = parse_line(line)
+        if (@predicted = model.predict(sample)) != (@actual = output_lines[index].to_i)
+          @failed_line = index + 1
+          break
+        end
+      end
+      @failed_line.nil?
+    end
+    
+    failure_message_for_should do |actual|
+      "Incorrect prediction on line #{@failed_line}, expected #{@actual} got #{@predicted}"
+    end
+    
+    chain :for_input do |input|
+      @input = input
+    end
+  end
+  
+end

data/spec/model_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+
+
+describe(RubyLinear::Model) do
+  
+  let :test_vector do#first line from dna.scale.t
+    {6 => 1, 7 => 1, 11 => 1, 18 => 1, 20 => 1, 24 => 1, 27 => 1, 30 => 1, 33 => 1, 34 => 1, 38 => 1, 42 => 1, 45 => 1, 47 => 1, 53 => 1, 60 => 1, 61 => 1, 65 => 1, 69 => 1, 70 => 1, 75 => 1, 78 => 1, 79 => 1, 84 => 1, 87 => 1, 88 => 1, 92 => 1, 99 => 1, 101 => 1, 103 => 1, 108 => 1, 110 => 1, 112 => 1, 119 => 1, 123 => 1, 124 => 1, 128 => 1, 131 => 1, 134 => 1, 137 => 1, 139 => 1, 142 => 1, 147 => 1, 149 => 1, 156 => 1, 157 => 1, 161 => 1, 164 => 1, 166 => 1, 171 => 1, 173 => 1, 180 => 1}
+  end
+  
+  describe('load_file') do
+    before(:each) do
+      @model = RubyLinear::Model.load_file(File.dirname(__FILE__) + '/fixtures/dna.dat')
+    end
+    
+    it 'should load the model from the path' do
+      @model.bias.should == 1
+      @model.class_count.should == 3
+      @model.feature_count.should == 180
+      @model.labels.should == [3,1,2]
+    end
+    
+    it 'should be able to predict' do
+      @model.predict(test_vector).should == 3
+    end
+  end
+  
+  describe('predict_values') do
+    before(:each) do
+      @model = RubyLinear::Model.load_file(File.dirname(__FILE__) + '/fixtures/dna.dat')
+    end
+    
+    it 'should return the class and a hash of labels to values' do
+      label, values = @model.predict_values(test_vector)
+      
+      label.should == 3
+      values.keys.should =~ [1,2,3]
+      values[3].should be_within(0.001).of(4.178)
+      values[2].should be_within(0.001).of(-8.477)
+      values[1].should be_within(0.001).of(-3.568)
+    end
+  end
+  describe('new') do
+    let(:problem) {RubyLinear::Problem.load_file(File.dirname(__FILE__) + '/fixtures/dna.scale.txt', 1.0)}
+    it 'should train the model from the parameters and the problem' do
+      m = RubyLinear::Model.new(problem, :solver => RubyLinear::L1R_L2LOSS_SVC)
+      m.predict(test_vector).should == 3
+    end
+    
+    context 'when unknwon options are presented' do
+      it 'should raise argument error' do
+        expect { RubyLinear::Model.new(problem, :solver => RubyLinear::L1R_L2LOSS_SVC, :bogus_option => true) }.to raise_error(ArgumentError)
+        
+      end
+      
+    end
+  end
+end

data/spec/problem_spec.rb

@@ -0,0 +1,83 @@
+require 'spec_helper'
+
+
+describe(RubyLinear::Problem) do
+  
+  describe 'load_file' do
+    it 'should load a libsvm format file' do
+      problem = RubyLinear::Problem.load_file(File.dirname(__FILE__) + '/fixtures/dna.scale.txt', -1)
+      problem.bias.should == -1
+      
+      problem.l.should == 2000
+      problem.n.should == 180
+      
+      #line 1 from the file
+      problem.feature_vector(0).should == [[2,1], [7,1], [12,1], [15,1], [17,1], [23,1], [26,1], [28,1], [33,1], [34,1], [40,1], [45,1], [47,1], [50,1], [52,1], [58,1], [63,1], [64,1], [67,1], [72,1], [73,1], [76,1], [80,1], [83,1], [85,1], [88,1], [91,1], [95,1], [97,1], [101,1], [113,1], [120,1], [122,1], [126,1], [132,1], [138,1], [144,1], [145,1], [150,1], [151,1], [154,1], [160,1], [163,1], [170,1], [172,1], [177,1], [178,1]] 
+
+      problem.labels.length.should == 2000
+    end
+    
+    it 'should add 1 to the feature count if the bias is > 0' do
+      problem = RubyLinear::Problem.load_file(File.dirname(__FILE__) + '/fixtures/dna.scale.txt', 1)
+      
+      problem.l.should == 2000
+      problem.n.should == 181
+
+      problem.feature_vector(0).should == [[2,1], [7,1], [12,1], [15,1], [17,1], [23,1], [26,1], [28,1], [33,1], [34,1], [40,1], [45,1], [47,1], [50,1], [52,1], [58,1], [63,1], [64,1], [67,1], [72,1], [73,1], [76,1], [80,1], [83,1], [85,1], [88,1], [91,1], [95,1], [97,1], [101,1], [113,1], [120,1], [122,1], [126,1], [132,1], [138,1], [144,1], [145,1], [150,1], [151,1], [154,1], [160,1], [163,1], [170,1], [172,1], [177,1], [178,1],[181,1]] 
+
+    end
+  end
+  
+  describe 'destroy' do
+  
+    it 'should release associated memory' do
+      problem = RubyLinear::Problem.load_file(File.dirname(__FILE__) + '/fixtures/dna.scale.txt', 1)
+      problem.destroyed?.should be_false
+      problem.destroy!
+      problem.destroyed?.should be_true
+    end
+      
+  end
+  
+  describe 'new' do
+    before(:each) do
+      @samples = [ {2=>0.1,3=>0.3,4=>-1.2},{1=>0.4},{2=>0.1, 4=>1.4,5=>0.5},{1 => -0.1, 2=> -0.2, 3=>0.1,4=>1.1,5=>0.1}]
+      @labels = [2,1,2,3]
+      @max_feature = 5
+    end
+    
+    context 'when max feature is inconsistent with the samples' do
+      it 'should raise argument error' do
+        expect {RubyLinear::Problem.new(@labels, @samples, -1, 3)}.to raise_error(ArgumentError, /inconsistent/)
+      end
+    end
+    
+    context 'when samples and labels are of different length' do
+      it 'should raise argument error' do
+        expect {RubyLinear::Problem.new([1,2,3], @samples, -1, @max_feature)}.to raise_error(ArgumentError, /different length/)
+      end
+    end
+    context 'when the bias is < 0' do
+      it 'should create a new problem' do
+        problem = RubyLinear::Problem.new(@labels, @samples, -1, @max_feature)
+        
+        problem.l.should == 4
+        problem.n.should == 5
+        problem.bias.should == -1
+        problem.labels.should == [2,1,2,3]
+        problem.feature_vector(0).should == [[2,0.1], [3,0.3], [4,-1.2]]
+      end
+    end
+    
+    context 'when the bias is > 0' do
+      it 'should add a  bias term to each vextor' do
+        problem = RubyLinear::Problem.new(@labels, @samples, 1.0, @max_feature)
+        #all the feature vectors should also be padded with
+        problem.l.should == 4
+        problem.n.should == 6
+        problem.bias.should == 1
+        problem.feature_vector(0).should == [[2,0.1], [3,0.3], [4,-1.2], [6,1]]
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'bundler/setup'
+
+$: << File.dirname(__FILE__) + '/../ext'
+
+require 'ruby_linear'
+
+RSpec.configure do |config|
+  
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: ruby_linear
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Frederick Cheung
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-29 00:00:00.000000000 Z
+dependencies: []
+description: ! 'Ruby wrapper for LIBLINEAR, a library for large linear classification '
+email: frederick.cheung@gmail.com
+executables: []
+extensions:
+- ext/extconf.rb
+extra_rdoc_files: []
+files:
+- COPYING
+- AUTHORS
+- README.markdown
+- Rakefile
+- ext/extconf.rb
+- ext/blas.h
+- ext/blasp.h
+- ext/linear.h
+- ext/tron.h
+- ext/linear.cpp
+- ext/rubylinear.cpp
+- ext/tron.cpp
+- ext/daxpy.c
+- ext/ddot.c
+- ext/dnrm2.c
+- ext/dscal.c
+- lib/ruby_linear.rb
+- spec/fixtures/dna.dat
+- spec/fixtures/dna.out
+- spec/fixtures/dna.scale.t
+- spec/fixtures/dna.scale.txt
+- spec/integration_spec.rb
+- spec/model_spec.rb
+- spec/problem_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/fcheung/rubylinear
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Ruby wrapper for LIBLINEAR
+test_files: []