libsvm 接口 [供自己的C/C++程序训练预测用]

 

#pragma once
/*
 * lp_svm.h : a svm class use libSVM to train and predict data.
 */
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <ctime>
#include <vector>
#include <iterator>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <float.h>
#include <string.h>
#include <stdarg.h>
#include <limits.h>
#include <locale.h>
#include <errno.h>
#include "/home/lvpeng/src/lpinclude/svm/svm.h"
using namespace std;

#define Malloc(type,n) (type *)malloc((n)*sizeof(type))

/*
 * Class: lpCSVM
 * Func: read & predict & train data
 * Author: Peng Lv
 * Data: 12/04/2012
 */
class lpCSVM{
public:
	struct svm_parameter m_param;
	struct svm_problem m_prob;
	struct svm_model *m_model;
	struct svm_node *m_xSpace;

	int m_crossValid;
	int m_nrFold;
	int m_dim;

	char* line;
	int max_line_len;
	int max_nr_attr;
	int m_predict_probability;
public:
	lpCSVM(){};
	~lpCSVM();

	char* readLine(FILE *);

	/////////////////////////////////
	//   interface functions 
	/////////////////////////////////
	void initSVM();
	void readProblem(const char*);
	void trainModel(char* input, char * output);
	int loadModel(char* );
	void predictFile(FILE* , FILE* );
	double predictX(vector<double>& );
};