hexgnu-libsvm-ruby-swig 0.1.0

data/ext/svm.h

@@ -0,0 +1,76 @@
+#ifndef _LIBSVM_H
+#define _LIBSVM_H
+
+#define LIBSVM_VERSION 290
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+extern int libsvm_version;
+
+struct svm_node
+{
+	int index;
+	double value;
+};
+
+struct svm_problem
+{
+	int l;
+	double *y;
+	struct svm_node **x;
+};
+
+enum { C_SVC, NU_SVC, ONE_CLASS, EPSILON_SVR, NU_SVR };	/* svm_type */
+enum { LINEAR, POLY, RBF, SIGMOID, PRECOMPUTED }; /* kernel_type */
+
+struct svm_parameter
+{
+	int svm_type;
+	int kernel_type;
+	int degree;	/* for poly */
+	double gamma;	/* for poly/rbf/sigmoid */
+	double coef0;	/* for poly/sigmoid */
+
+	/* these are for training only */
+	double cache_size; /* in MB */
+	double eps;	/* stopping criteria */
+	double C;	/* for C_SVC, EPSILON_SVR and NU_SVR */
+	int nr_weight;		/* for C_SVC */
+	int *weight_label;	/* for C_SVC */
+	double* weight;		/* for C_SVC */
+	double nu;	/* for NU_SVC, ONE_CLASS, and NU_SVR */
+	double p;	/* for EPSILON_SVR */
+	int shrinking;	/* use the shrinking heuristics */
+	int probability; /* do probability estimates */
+};
+
+struct svm_model *svm_train(const struct svm_problem *prob, const struct svm_parameter *param);
+void svm_cross_validation(const struct svm_problem *prob, const struct svm_parameter *param, int nr_fold, double *target);
+
+int svm_save_model(const char *model_file_name, const struct svm_model *model);
+struct svm_model *svm_load_model(const char *model_file_name);
+
+int svm_get_svm_type(const struct svm_model *model);
+int svm_get_nr_class(const struct svm_model *model);
+void svm_get_labels(const struct svm_model *model, int *label);
+double svm_get_svr_probability(const struct svm_model *model);
+
+void svm_predict_values(const struct svm_model *model, const struct svm_node *x, double* dec_values);
+double svm_predict(const struct svm_model *model, const struct svm_node *x);
+double svm_predict_probability(const struct svm_model *model, const struct svm_node *x, double* prob_estimates);
+
+void svm_destroy_model(struct svm_model *model);
+void svm_destroy_param(struct svm_parameter *param);
+
+const char *svm_check_parameter(const struct svm_problem *prob, const struct svm_parameter *param);
+int svm_check_probability_model(const struct svm_model *model);
+
+extern void (*svm_print_string) (const char *);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* _LIBSVM_H */

data/lib/svm.rb

@@ -0,0 +1,124 @@
+require File.join(File.dirname(__FILE__), 'libsvm')
+include Libsvm
+
+%w[./libsvm/problem ./libsvm/model ./libsvm/parameter].each do |r|
+  require File.join(File.dirname(__FILE__), r)
+end
+
+# TODO: these are super weird should be inside of a module; extend self; end...
+
+
+def _int_array(seq)
+  size = seq.size
+  array = new_int(size)
+  i = 0
+  for item in seq
+    int_setitem(array,i,item)
+    i = i + 1
+  end
+  return array
+end
+
+def _double_array(seq)
+  size = seq.size
+  array = new_double(size)
+  i = 0
+  for item in seq
+    double_setitem(array,i,item)
+    i = i + 1
+  end
+  return array
+end
+
+def _free_int_array(x)
+  if !x.nil? and !x.empty?
+    delete_int(x)
+  end
+end
+
+def _free_double_array(x)
+  if !x.nil? and !x.empty?
+    delete_double(x)
+  end
+end
+
+def _int_array_to_list(x,n)
+  list = []
+   (0..n-1).each {|i| list << int_getitem(x,i) }
+  return list
+end
+
+def _double_array_to_list(x,n)
+  list = []
+   (0..n-1).each {|i| list << double_getitem(x,i) }
+  return list
+end    
+
+def _convert_to_svm_node_array(x)
+  # Make index array 
+  iter_range = x.each_index.to_a
+  
+  data = svm_node_array(iter_range.length + 1)
+  svm_node_array_set(data, iter_range.length, -1, 0)
+  
+  iter_range.each do |k|
+    svm_node_array_set(data, k, k, x[k])
+  end
+  
+  data
+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)
+    end
+
+    data
+  end
+end
+
+
+def cross_validation(prob, param, fold)
+  if param.gamma == 0
+    param.gamma = 1.0/prob.maxlen
+  end
+  dblarr = new_double(prob.size)
+  svm_cross_validation(prob.prob, param.param, fold, dblarr)
+  ret = _double_array_to_list(dblarr, prob.size)
+  delete_double(dblarr)
+  return ret
+end
+
+def read_file filename
+  labels = []
+  samples = []
+  max_index = 0
+
+  f = File.open(filename)
+  f.each do |line|
+    elems = line.split
+    sample = {}
+    for e in elems[1..-1]
+       points = e.split(":")
+       sample[points[0].to_i] = points[1].to_f
+       if points[0].to_i < max_index
+          max_index = points[0].to_i
+       end
+    end
+    labels << elems[0].to_i
+    samples << sample
+  end
+  puts "#{filename}: #{samples.size} samples loaded."
+  return labels,samples
+end
+

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification 
+name: hexgnu-libsvm-ruby-swig
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Tom Zeng
+- Matthew Kirk
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-17 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 49
+        segments: 
+        - 1
+        - 8
+        - 3
+        version: 1.8.3
+  type: :development
+  version_requirements: *id001
+description: Ruby wrapper of LIBSVM using SWIG
+email: matt@matthewkirk.com
+executables: []
+
+extensions: 
+- ext/extconf.rb
+extra_rdoc_files: 
+- History.txt
+- Manifest.txt
+- README.rdoc
+files: 
+- History.txt
+- COPYING
+- AUTHORS
+- Manifest.txt
+- README.rdoc
+- Rakefile
+- lib/svm.rb
+- ext/libsvm_wrap.cxx
+- ext/svm.cpp
+- ext/svm.h
+- ext/extconf.rb
+has_rdoc: false
+homepage: http://www.matthewkirk.com
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README.rdoc
+require_paths: 
+- lib
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.0
+signing_key: 
+specification_version: 2
+summary: Ruby wrapper fork of Tom Zeng's libsvm wrapper
+test_files: []
+