shogun_libsvm.h

Go to the documentation of this file.

/*
 * Copyright (c) 2000-2009 Chih-Chung Chang and Chih-Jen Lin
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 *
 * 3. Neither name of copyright holders nor the names of its contributors
 * may be used to endorse or promote products derived from this software
 * without specific prior written permission.
 *
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Shogun specific adjustments (w) 2006-2009 Soeren Sonnenburg
 */

#ifndef DOXYGEN_SHOULD_SKIP_THIS

#ifndef _LIBSVM_H
#define _LIBSVM_H

#include <shogun/kernel/Kernel.h>

namespace shogun
{
struct svm_node
{
    int32_t index;
};

struct svm_problem
{


public:

    svm_problem() {
        y = NULL;
        x = NULL;
        C = NULL;
        pv = NULL;
    }


    int32_t l;
    float64_t *y;
    struct svm_node **x;
    float64_t *C;
    float64_t *pv;

};

enum { C_SVC=1, NU_SVC=2, NU_MULTICLASS_SVC=3, ONE_CLASS=4, EPSILON_SVR=5, NU_SVR=6 };  /* svm_type */
enum { LINEAR, POLY, RBF, SIGMOID, PRECOMPUTED }; /* kernel_type */

struct svm_parameter
{
    int32_t svm_type;
    int32_t kernel_type;
    shogun::CKernel* kernel;
    int32_t degree;
    float64_t gamma;
    float64_t coef0;

    /* these are for training only */
    float64_t cache_size;
    float64_t max_train_time;
    float64_t eps;
    float64_t C;
    int32_t nr_class;
    int32_t nr_weight;
    int32_t *weight_label;
    float64_t* weight;
    float64_t nu;
    float64_t p;
    int32_t shrinking;
    bool use_bias;
};

struct svm_model
{
    svm_parameter param;
    int32_t nr_class;
    int32_t l;
    svm_node **SV;
    float64_t **sv_coef;
    float64_t *rho;
    float64_t *normwcw;

    // for classification only

    int32_t *label;
    int32_t *nSV;
    // XXX
    int32_t free_sv;
    float64_t objective;
};



struct svm_model *svm_train(
    const struct svm_problem *prob, const struct svm_parameter *param);

float64_t svm_predict(const struct svm_model *model, const struct svm_node *x);

void svm_destroy_model(struct svm_model *model);

const char *svm_check_parameter(const struct svm_problem *prob, const struct svm_parameter *param);
}
#endif /* _LIBSVM_H */

#endif // DOXYGEN_SHOULD_SKIP_THIS