KernelMeanMatching.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 (W) 2012 Sergey Lisitsyn
 */

#ifndef KERNELMEANMATCHING_H_
#define KERNELMEANMATCHING_H_

#include <shogun/lib/config.h>
#ifdef USE_GPL_SHOGUN

#include <shogun/base/SGObject.h>
#include <shogun/kernel/Kernel.h>

namespace shogun
{

class CKernelMeanMatching: public CSGObject
{
public:

    CKernelMeanMatching();

    CKernelMeanMatching(CKernel* kernel, SGVector<index_t> training_indices, SGVector<index_t> test_indices);

    CKernel* get_kernel() const { SG_REF(m_kernel); return m_kernel; }
    void set_kernel(CKernel* kernel) { SG_REF(kernel); SG_UNREF(m_kernel); m_kernel = kernel; }
    SGVector<index_t> get_training_indices() const { return m_training_indices; }
    void set_training_indices(SGVector<index_t> training_indices) { m_training_indices = training_indices; }
    SGVector<index_t> get_test_indices() const { return m_test_indices; }
    void set_test_indices(SGVector<index_t> test_indices) { m_test_indices = test_indices; }

    SGVector<float64_t> compute_weights();

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

protected:

    CKernel* m_kernel;
    SGVector<index_t> m_training_indices;
    SGVector<index_t> m_test_indices;
};

}
#endif //USE_GPL_SHOGUN
#endif