MulticlassSOLabels.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) 2013 Thoralf Klein
 * Written (W) 2012 Fernando José Iglesias García
 * Copyright (C) 2012 Fernando José Iglesias García
 */

#ifndef _MULTICLASS_SO_LABELS__H__
#define _MULTICLASS_SO_LABELS__H__

#include <shogun/lib/config.h>

#include <shogun/labels/StructuredLabels.h>
#include <shogun/lib/SGVector.h>
#include <shogun/lib/StructuredData.h>
#include <shogun/lib/StructuredDataTypes.h>

namespace shogun
{

class CStructuredLabels;
class CMulticlassSOLabels;

struct CRealNumber : public CStructuredData
{
    STRUCTURED_DATA_TYPE(SDT_REAL);

    CRealNumber(float64_t val) : CStructuredData(), value(val) { }

    static CRealNumber* obtain_from_generic(CStructuredData* base_data)
    {
        if ( base_data->get_structured_data_type() == SDT_REAL )
            return (CRealNumber*) base_data;
        else
            SG_SERROR("base_data must be of dynamic type CRealNumber\n")

        return NULL;
    }

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

    float64_t value;
};

class CMulticlassSOLabels : public CStructuredLabels
{
    public:
        CMulticlassSOLabels();

        CMulticlassSOLabels(int32_t num_labels);

        CMulticlassSOLabels(SGVector< float64_t > const src);

        virtual ~CMulticlassSOLabels();

        inline int32_t get_num_classes() { return m_num_classes; }

        virtual void add_label(CStructuredData* label);

        virtual CStructuredData* get_label(int32_t idx);

        virtual bool set_label(int32_t idx, CStructuredData* label);

        virtual int32_t get_num_labels() const;

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

    private:
        void init();

    private:
        int32_t m_num_classes;

        SGVector< float64_t > m_labels_vector;
        int32_t m_num_labels_set;

}; /* CMulticlassSOLabels */

} /* namespace shogun */

#endif /* _MULTICLASS_SO_LABELS__H__ */