MultitaskLinearMachine.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.
 *
 * Copyright (C) 2012 Sergey Lisitsyn
 */

#ifndef  MULTITASKMACHINE_H_
#define  MULTITASKMACHINE_H_

#include <shogun/lib/config.h>
#ifdef USE_GPL_SHOGUN
#include <shogun/machine/LinearMachine.h>
#include <shogun/transfer/multitask/TaskRelation.h>
#include <shogun/transfer/multitask/TaskGroup.h>
#include <shogun/transfer/multitask/TaskTree.h>
#include <shogun/transfer/multitask/Task.h>

#include <vector>
#include <set>

namespace shogun
{
class CMultitaskLinearMachine : public CLinearMachine
{

    public:
        CMultitaskLinearMachine();

        CMultitaskLinearMachine(
             CDotFeatures* training_data,
             CLabels* training_labels, CTaskRelation* task_relation);

        virtual ~CMultitaskLinearMachine();

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

        int32_t get_current_task() const;

        void set_current_task(int32_t task);

        virtual SGVector<float64_t> get_w() const;

        virtual void set_w(const SGVector<float64_t> src_w);

        virtual void set_bias(float64_t b);

        virtual float64_t get_bias();

        CTaskRelation* get_task_relation() const;

        void set_task_relation(CTaskRelation* task_relation);

        virtual bool supports_locking() const { return true; }

        virtual void post_lock(CLabels* labels, CFeatures* features_);

        virtual bool train_locked(SGVector<index_t> indices);

        virtual CBinaryLabels* apply_locked_binary(SGVector<index_t> indices);

        virtual float64_t apply_one(int32_t i);

    protected:

        virtual SGVector<float64_t> apply_get_outputs(CFeatures* data=NULL);

        virtual bool train_machine(CFeatures* data=NULL);

        virtual bool train_locked_implementation(SGVector<index_t>* tasks);

        SGVector<index_t>* get_subset_tasks_indices();

    private:

        void register_parameters();

    protected:

        int32_t m_current_task;

        CTaskRelation* m_task_relation;

        SGMatrix<float64_t> m_tasks_w;

        SGVector<float64_t> m_tasks_c;

        std::vector< std::set<index_t> > m_tasks_indices;

};
}
#endif //USE_GPL_SHOGUN
#endif