LinearStructuredOutputMachine.cpp

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 José Iglesias García
 * Copyright (C) 2012 Fernando José Iglesias García
 */

#include <shogun/machine/LinearStructuredOutputMachine.h>
#include <shogun/structure/MulticlassSOLabels.h>

using namespace shogun;

CLinearStructuredOutputMachine::CLinearStructuredOutputMachine()
: CStructuredOutputMachine()
{
    register_parameters();
}

CLinearStructuredOutputMachine::CLinearStructuredOutputMachine(
        CStructuredModel*  model, 
        CLossFunction*     loss, 
        CStructuredLabels* labs)
: CStructuredOutputMachine(model, loss, labs)
{
    register_parameters();
}

CLinearStructuredOutputMachine::~CLinearStructuredOutputMachine()
{
}

void CLinearStructuredOutputMachine::set_features(CFeatures* f)
{
    m_model->set_features(f);
}

CFeatures* CLinearStructuredOutputMachine::get_features() const
{
    return m_model->get_features();
}

SGVector< float64_t > CLinearStructuredOutputMachine::get_w() const
{
    return m_w;
}

CStructuredLabels* CLinearStructuredOutputMachine::apply_structured(CFeatures* data)
{
    if (data)
    {
        set_features(data);
    }

    CStructuredLabels* out;
    CFeatures* model_features = this->get_features();
    if (!model_features)
    {
        out = new CStructuredLabels();
    }
    else
    {
        out = new CStructuredLabels(model_features->get_num_vectors());
        for ( int32_t i = 0 ; i < model_features->get_num_vectors() ; ++i )
        {
            CResultSet* result = m_model->argmax(m_w, i, false);
            out->add_label(result->argmax);

            SG_UNREF(result);
        }
    }
    SG_UNREF(model_features);
    return out;
}

void CLinearStructuredOutputMachine::register_parameters()
{
    SG_ADD(&m_w, "m_w", "Weight vector", MS_NOT_AVAILABLE);
}