MultilabelModel.h

Go to the documentation of this file.

/*
 * This software is distributed under BSD 3-clause license (see LICENSE file).
 *
 * Copyright (C) 2014 Abinash Panda
 * Written (W) 2014 Abinash Panda
 */

#ifndef _MULTILABEL_MODEL__H__
#define _MULTILABEL_MODEL__H__

#include <shogun/lib/config.h>
#include <shogun/structure/StructuredModel.h>

namespace shogun
{

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

    CMultilabelModel(CFeatures * features, CStructuredLabels * labels);

    virtual ~CMultilabelModel();


    virtual CStructuredLabels * structured_labels_factory(int32_t num_labels = 0);

    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 set_misclass_cost(float64_t false_positive,
                                   float64_t false_negative);

    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 const char * get_name() const
    {
        return "MultilabelModel";
    }

private:
    float64_t m_false_positive;
    float64_t m_false_negative;
    int32_t m_num_classes;

private:
    void init();

    float64_t delta_loss(SGVector<float64_t> y1, SGVector<float64_t> y2);
    float64_t delta_loss(float64_t y1, float64_t y2);

    SGVector<int32_t> to_sparse(SGVector<float64_t> dense_vector,
                                float64_t d_true, float64_t d_false);

}; /* class CMultilabelModel */

} /* namespace shogun */

#endif /* _MULTILABEL_MODEL__H__ */