libsvmffi 0.0.1 → 0.0.2

data/lib/libsvmffi/model.rb

@@ -11,6 +11,8 @@ module Libsvmffi
     
     attr_accessor :nodes, :x
 
+    TMP_MODEL_FILE = "/tmp/libsvm_model.out"
+
     def initialize(options = {})
 
       @parameters = Parameters.new      
@@ -63,7 +65,7 @@ module Libsvmffi
       @problem[:l] = @examples.length
       @problem[:y] = FFI::MemoryPointer.new(:double, @problem[:l])
       @x = FFI::MemoryPointer.new(:pointer, @problem[:l])
-      @problem[:x] = x.address #FFI::MemoryPointer.new(:pointer, @problem[:l])
+      @problem[:x] = @x.address #FFI::MemoryPointer.new(:pointer, @problem[:l])
       @x_space = FFI::MemoryPointer.new(Node, @elements) 
 
       y = @examples.map {|e| e.keys.first}
@@ -104,11 +106,45 @@ module Libsvmffi
     end
     
 
+    def marshal_dump
+      
+        d = Marshal.dump @labels
+        d += "::::"
+        d += Marshal.dump @features
+      if !@svm_model.nil?
+        # TODO surely there is a better way to do this.
+        save_raw
+        raw_str = File.open(TMP_MODEL_FILE, "r").read
+        d += "::::"  
+        d += raw_str
+      end
+      
+      return d
+    end
+
+    def marshal_load(str)
+      @labels, @features, raw_model = str.split("::::")
+      
+      @labels = Marshal.load @labels
+      @features = Marshal.load @features
+      
+      File.open(TMP_MODEL_FILE, "w") {|f| f.write raw_model}
+      restore_raw
+    end
+  
     #
     # Save to file
     #
-    def save(filename = "model.out")
-      Libsvmffi.svm_save_model FFI::MemoryPointer.from_string(filename), @svm_model
+    def save_raw
+      Libsvmffi.svm_save_model FFI::MemoryPointer.from_string(TMP_MODEL_FILE), @svm_model
+    end
+
+    #
+    # Restore a model from a file. 
+    #   NOTE: currently does not restore pre-train data
+    #
+    def restore_raw
+      @svm_model = Libsvmffi.svm_load_model FFI::MemoryPointer.from_string(TMP_MODEL_FILE)
     end
     
     #

metadata

@@ -2,7 +2,7 @@
 name: libsvmffi
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Robert Berry
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-01 00:00:00 Z
+date: 2011-07-10 00:00:00 Z
 dependencies: []
 
 description: SVM operations.
@@ -52,10 +52,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.7.2
+rubygems_version: 1.8.5
 signing_key: 
 specification_version: 3
 summary: Libsvm ffi bindings
 test_files: []
 
-has_rdoc: false