hexgnu-libsvm-ruby-swig 0.1.2 → 0.1.3

data/lib/hex-svm.rb

@@ -71,16 +71,25 @@ end
 module SVM
   extend self
   
-  def convert_to_svm_node_array(x, max = 30) 
-    # X is an array of indicies
-    # Make index array 
-    # iter_range = x.sort
-
-    data = svm_node_array(max + 1)
-    svm_node_array_set(data, max, -1, 0)
-
-    x.sort.each do |k|
-      svm_node_array_set(data, k, k, 1)
+  def convert_to_svm_node_array(indicies, max) 
+    # Make index array
+    # x = indexes_to_array(indicies, max)
+    # iter_range = x.each_index.to_a
+    
+    data = svm_node_array(indicies.length + 1)
+    svm_node_array_set(data, indicies.length, -1, 0)
+    
+    # max.times do |i|
+    #   # Set to zero if not in indicies
+    #   if indicies.include?(i)
+    #     svm_node_array_set(data, i, i, 1)
+    #   else
+    #     svm_node_array_set(data, i, i, 0)
+    #   end
+    # end
+    
+    indicies.sort.each_with_index do |idx, i|
+      svm_node_array_set(data, i, idx, 1)
     end
 
     data

data/lib/libsvm/model.rb

@@ -16,7 +16,7 @@ module SVM
         end
         msg = svm_check_parameter(prob.prob,param.param)
         raise ::ArgumentError, msg if msg
-        @model = svm_train(prob.prob,param.param)
+        @model = svm_train(prob.prob, param.param)
       end
     
       #setup some classwide variables
@@ -29,11 +29,10 @@ module SVM
       delete_int(intarr)
       #check if valid probability model
       @probability = svm_check_probability_model(@model)
-
     end
   
-    def predict(x)
-      data = SVM.convert_to_svm_node_array(x)
+    def predict(x, max = x.max)
+      data = SVM.convert_to_svm_node_array(x, max)
       ret = svm_predict(@model,data)
       svm_node_array_destroy(data)
       return ret
@@ -54,7 +53,7 @@ module SVM
     def predict_values_raw(x)
       #convert x into svm_node, allocate a double array for return
       n = (@nr_class*(@nr_class-1)/2).floor
-      data = _convert_to_svm_node_array(x)
+      data = SVM.convert_to_svm_node_array(x)
       dblarr = new_double(n)
       svm_predict_values(@model, data, dblarr)
       ret = _double_array_to_list(dblarr, n)
@@ -101,7 +100,7 @@ module SVM
       end
     
       #convert x into svm_node, alloc a double array to receive probabilities
-      data = _convert_to_svm_node_array(x)
+      data = SVM.convert_to_svm_node_array(x)
       dblarr = new_double(@nr_class)
       pred = svm_predict_probability(@model, data, dblarr)
       pv = _double_array_to_list(dblarr, @nr_class)

data/lib/libsvm/problem.rb

@@ -1,9 +1,11 @@
 module SVM
   class Problem
     attr_accessor :prob, :maxlen, :size
+    attr_reader :max
     
     def initialize(y, *x)
       #assert y.size == x.size
+      @max = x.flatten.max #aggh hate doing this...
       @prob = prob = Svm_problem.new 
       @size = size = y.size
     
@@ -14,12 +16,10 @@ module SVM
     
       @x_matrix = x_matrix = svm_node_matrix(size)
       @data = []
-      @maxlen = 0
+      @maxlen = max
       x.each_with_index do |row, i|
-        #data = _convert_to_svm_node_array(row)
-        data = SVM.convert_to_svm_node_array(row)
-        @data << data
-        svm_node_matrix_set(x_matrix, i, data)
+        @data << SVM.convert_to_svm_node_array(row, max)
+        svm_node_matrix_set(x_matrix, i, @data.last)
         @maxlen = [@maxlen, row.size].max
       end
     

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: hexgnu-libsvm-ruby-swig
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: 
   segments: 
   - 0
   - 1
-  - 2
-  version: 0.1.2
+  - 3
+  version: 0.1.3
 platform: ruby
 authors: 
 - Tom Zeng