github.com/cyrusf/libsvm-go

libsvm-go: Support Vector Machine

This is a full port of LIBSVM in the Go programming language. LIBSVM is a suite of tools and an API library for support vector classification, regression, and distribution estimation. This port implements the libsvm library in the form of a Go package called libSvm. It also implements the svm-train and svm-predict command line tools.

This port has no external package dependencies, and uses only the native standard library.

It was based on ewalker544's project, and this one was added a new funtion of load model from a byte slice. It is useful to build project into a single binary file.

Installation

go get github.com/CyrusF/libsvm-go
make

Compatibility Notes

I have tried to make the Go implementation of svm-train and svm-predict plug-in compatibile with the original LIBSVM 3.18 distribution. This is to allow you to use the other tools available in the original distribution, like easy.py and grid.py.

svm-predict should be 100% plug-in compatibile. However, svm-train is plug-in compatible with one exception. The exception is the parameter weight flag used in the command. In this implementation, the flag is

-w i,weight : set the parameter C of class i to weight*C, for C-SVC (default 1)

For full documentation of the svm-train and svm-predict commands, please refer to the original LIBSVM web site.

API Example

Training

import "github.com/CyrusF/libsvm-go"
    
param := libSvm.NewParameter()      // Create a parameter object with default values
param.KernelType = libSvm.POLY      // Use the polynomial kernel
    
model := libSvm.NewModel(param)     // Create a model object from the parameter attributes
    
// Create a problem specification from the training data and parameter attributes
problem, err := libSvm.NewProblem("a9a.train", param) 
    
model.Train(problem)                // Train the model from the problem specification
    
model.Dump("a9a.model")             // Dump the model into a user-specified file

Predicting

import "github.com/CyrusF/libsvm-go"
    
// Create a model object from the model file generated from training
model := libSvm.NewModelFromFile("a9a.model")  
    
x := make(map[int]float64)
// Populate x with the test vector
    
predictLabel := model.Predict(x)    // Predicts a float64 label given the test vector 

Another way to load model

import "github.com/CyrusF/libsvm-go"
    
// Create a model object from the io.Reader or byte slice generated from training
byteReader := bytes.NewReader([]byte(""))
model := libSvm.NewModelFromFileStream(byteReader)  
    
x := make(map[int]float64)
// Populate x with the test vector
    
predictLabel := model.Predict(x)    // Predicts a float64 label given the test vector