您的位置:首页 > 编程语言 > C语言/C++

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

2012-04-14 00:32 363 查看
#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>& );
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: