svm_tree.h

Go to the documentation of this file.

/*
 * This work is licensed under a Creative Commons 
 * Attribution-Noncommercial-Share Alike 3.0 United States License.
 * 
 *    http://creativecommons.org/licenses/by-nc-sa/3.0/us/
 * 
 * You are free:
 * 
 *    to Share - to copy, distribute, display, and perform the work
 *    to Remix - to make derivative works
 * 
 * Under the following conditions:
 * 
 *    Attribution. You must attribute the work in the manner specified by the
 *    author or licensor (but not in any way that suggests that they endorse you
 *    or your use of the work).
 * 
 *    Noncommercial. You may not use this work for commercial purposes.
 * 
 *    Share Alike. If you alter, transform, or build upon this work, you may
 *    distribute the resulting work only under the same or similar license to
 *    this one.
 * 
 * For any reuse or distribution, you must make clear to others the license
 * terms of this work. The best way to do this is by including this header.
 * 
 * Any of the above conditions can be waived if you get permission from the
 * copyright holder.
 * 
 * Apart from the remix rights granted under this license, nothing in this
 * license impairs or restricts the author's moral rights.
 */


#ifndef HAPLO_SVM_H
#define HAPLO_SVM_H


#include <config.h>


#include <stdlib.h>
#include <inttypes.h>

#if defined HAPLO_HAVE_LIBSVM_H
#include <libsvm.h>
#elif defined HAPLO_HAVE_SVM_H
#include <svm.h>
#endif

#include <jwsc/base/error.h>
#include <jwsc/vector/vector.h>
#include <jwsc/matrix/matrix.h>


typedef struct
{
    struct svm_model* svm;

    struct svm_problem* prob;
}
SVM_model;


void train_svm_model
(
    SVM_model**       model_out,
    const Vector_u32* labels, 
    const Matrix_i32* markers,
    double            cost,
    double            gamma
);


Error* predict_label_with_svm_model
(
    uint32_t*         label_out,
    double**          confidence_out,
    const struct svm_node*  markers,
    const SVM_model*  model
);


Error* predict_labels_with_svm_model
(
    Vector_u32**      labels_out,
    Vector_d**        confidence_out,
    const Matrix_i32* markers,
    const SVM_model*  model
);


Error* read_svm_model(SVM_model** model_out, const char* fname);


Error* write_svm_model(SVM_model* model, const char* fname);


Error* write_svm_model_training_data
(
    const Vector_u32* labels,
    const Matrix_i32* markers,
    const char*       fname
);


void free_svm_model(SVM_model* model);


typedef struct SVM_model_node
{
    const struct SVM_model_node* parent;

    uint32_t parent_label;

    struct SVM_model_node** subtrees[2];

    Vector_u32* labels[2];

    Vector_u32** altlabels[2];

    Vector_d* cost;

    Vector_d* gamma;

    uint32_t num_models; 

    SVM_model** models;

    char** model_fnames;
} 
SVM_model_node;


typedef  SVM_model_node  SVM_model_tree;


Error* train_svm_model_tree
(
    SVM_model_tree**    tree_out,
    const Vector_u32*   labels, 
    const Matrix_i32*   markers,
    const char*         tree_xml_fname,
    const char*         tree_dtd_fname
);


Error* predict_labels_with_svm_model_tree
(
    Vector_u32**          labels_out,
    Vector_d**            confidence_out,
    const Matrix_i32*     markers,
    const SVM_model_tree* tree
);


Error* read_svm_model_tree
(
    SVM_model_tree** tree_out,
    const char*      tree_xml_fname,
    const char*      tree_dtd_fname,
    const char*      model_dirname
);


Error* write_svm_model_tree
(
    const SVM_model_tree* tree,
    const char*              model_dirname
);


Error* write_svm_model_tree_training_data
(
    const Vector_u32* labels, 
    const Matrix_i32* markers,
    const char*       tree_xml_fname,
    const char*       tree_dtd_fname,
    const char*       data_dirname
);


void free_svm_model_tree(SVM_model_tree* tree);


#endif