HMSVMModel.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2012 Fernando Jose Iglesias Garcia
 * Copyright (C) 2012 Fernando Jose Iglesias Garcia
 */

#ifndef _HMSVM_MODEL__H__
#define _HMSVM_MODEL__H__

#include <shogun/structure/StructuredModel.h>
#include <shogun/structure/SequenceLabels.h>
#include <shogun/structure/StateModelTypes.h>
#include <shogun/structure/StateModel.h>

namespace shogun
{

enum EStateModelType;

class CHMSVMModel : public CStructuredModel
{
    public:
        CHMSVMModel();

        CHMSVMModel(CFeatures* features, CStructuredLabels* labels, EStateModelType smt, int32_t num_obs=0, bool use_plifs=false);

        virtual ~CHMSVMModel();

        virtual int32_t get_dim() const;

        virtual SGVector< float64_t > get_joint_feature_vector(int32_t feat_idx, CStructuredData* y);

        virtual CResultSet* argmax(SGVector< float64_t > w, int32_t feat_idx, bool const training = true);

        virtual float64_t delta_loss(CStructuredData* y1, CStructuredData* y2);

        virtual void init_primal_opt(
                float64_t regularization,
                SGMatrix< float64_t > & A,  SGVector< float64_t > a,
                SGMatrix< float64_t > B,  SGVector< float64_t > & b,
                SGVector< float64_t > lb, SGVector< float64_t > ub,
                SGMatrix < float64_t > & C);

        virtual bool check_training_setup() const;

        virtual int32_t get_num_aux() const;

        virtual int32_t get_num_aux_con() const;

        void set_use_plifs(bool use_plifs);

        virtual void init_training();

        SGMatrix< float64_t > get_transmission_weights() const;

        SGVector< float64_t > get_emission_weights() const;

        CStateModel* get_state_model() const;

        virtual const char* get_name() const { return "HMSVMModel"; }

    private:
        /* internal initialization */
        void init();

    private:
        int32_t m_num_obs;

        int32_t m_num_aux;

        CStateModel* m_state_model;

        SGMatrix< float64_t > m_transmission_weights;

        SGVector< float64_t > m_emission_weights;

        int32_t m_num_plif_nodes;

        CDynamicObjectArray* m_plif_matrix;

        bool m_use_plifs;

}; /* class CHMSVMModel */

} /* namespace shogun */

#endif /* _HMSVM_MODEL__H__ */