svmredlight 0.1.0

data/lib/svmredlight/document.rb

@@ -0,0 +1,23 @@
+module SVMLight
+  # A document is the Ruby representation of a DOC structure in SVMlight, it contains a
+  # queryid, a slackid, a costfactor ( c ) and a vector with feature numbers and their
+  # correspondent weights.
+  class Document
+    
+    # @param [Hash] vector a hash where the keys are feature numbers and the values its weights
+    # @param [Hash] opts the options coincide with SVMLight parameters to the create_example function, the default values for all the options are 0
+    # @option [:docnum] Numeric docum
+    # @option [:costfactor] Numeric costfactor
+    # @option [:slackid] Numeric slackid
+    # @option [:queryid] Numeric queryid
+    def self.new(vector, opts={})
+      opts.default = 0
+      docnum     = opts[:docnum]
+      costfactor = opts[:costfactor]
+      slackid    = opts[:slackid] 
+      queryid    = opts[:queryid] 
+
+      create(docnum, costfactor, slackid, queryid, vector.to_a)
+    end
+  end
+end

data/lib/svmredlight/model.rb

@@ -0,0 +1,22 @@
+module SVMLight
+  # A model is the product of training a SVM, once created it can take documents as inputs
+  # and act of them (by for instance classifying them). Models can also be read from files
+  # created by svm_learn.
+  class Model
+    TYPES = [:classification]
+
+    # Learns a model from a set of labeled documents.
+    # @param [Symbol] type, what kind of model is this, classification, regression, etc. for now the only valid value is classification.
+    # @param [Array] documents_and_lables documents and labels is an array of arrays where each inner array must have two elements, the first, a Document and the second a classification (normally +1 and  -1)
+    # @param [Hash] learn_params each key of learn_params is a string it that maps to a field of the LEARN_PARM struct in SVMLight
+    # @param [Hash] kernel_params each key of kernel_params is a string it that maps to a field of the KERNEL_PARM struct in SVMLight
+    # @param [Array|Nil] alphas an array of alpha values 
+    def self.new(type, documents_and_lables, learn_params, kernel_params, alphas = nil )
+      raise ArgumentError, "Supporte types are (for now) #{TYPES}" unless TYPES.include? type
+
+      learn_classification(documents_and_lables, learn_params, kernel_params, false, alphas)
+    end
+    
+    private_class_method :learn_classification
+  end
+end

data/lib/svmredlight.rb

@@ -0,0 +1,4 @@
+require File.dirname(__FILE__) + '/../ext/svmredlight'
+require  'svmredlight/model'
+require  'svmredlight/document'
+

data/svmredlight.gemspec

@@ -0,0 +1,73 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{svmredlight}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Camilo Lopez"]
+  s.date = %q{2011-09-11}
+  s.description = %q{Ruby interface to SVMLight}
+  s.email = %q{camilo@camilolopez.com}
+  s.extensions = ["ext/extconf.rb"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.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"
+  ]
+  s.homepage = %q{http://github.com/camilo/svmredlight}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.5.0}
+  s.summary = %q{Ruby interface to SVMLight}
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.6.0"])
+      s.add_development_dependency(%q<ZenTest>, [">= 0"])
+      s.add_development_dependency(%q<ruby-debug19>, [">= 0"])
+    else
+      s.add_dependency(%q<shoulda>, [">= 0"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.6.0"])
+      s.add_dependency(%q<ZenTest>, [">= 0"])
+      s.add_dependency(%q<ruby-debug19>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<shoulda>, [">= 0"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.6.0"])
+    s.add_dependency(%q<ZenTest>, [">= 0"])
+    s.add_dependency(%q<ruby-debug19>, [">= 0"])
+  end
+end
+