LinearTimeMMD.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) 2012 Heiko Strathmann
 */

#ifndef __LINEARTIMEMMD_H_
#define __LINEARTIMEMMD_H_

#include <shogun/statistics/KernelTwoSampleTestStatistic.h>
#include <shogun/kernel/Kernel.h>
#include <shogun/lib/external/libqp.h>

namespace shogun
{

class CFeatures;

class CLinearTimeMMD: public CKernelTwoSampleTestStatistic
{
public:
    CLinearTimeMMD();

    CLinearTimeMMD(CKernel* kernel, CFeatures* p_and_q, index_t q_start);

    CLinearTimeMMD(CKernel* kernel, CFeatures* p, CFeatures* q);

    virtual ~CLinearTimeMMD();

    virtual float64_t compute_statistic();

    virtual float64_t compute_p_value(float64_t statistic);

    virtual float64_t compute_threshold(float64_t alpha);

    virtual float64_t compute_variance_estimate();

#ifdef HAVE_LAPACK

    virtual void optimize_kernel_weights();

    void set_opt_max_iterations(index_t opt_max_iterations)
    {
        m_opt_max_iterations=opt_max_iterations;
    }

    void set_opt_epsilon(float64_t opt_epsilon) {
        m_opt_epsilon=opt_epsilon;
    }

    void set_opt_low_cut(float64_t opt_low_cut)
    {
        m_opt_low_cut=opt_low_cut;
    }

    void set_opt_regularization_eps(float64_t opt_regularization_eps)
    {
        m_opt_regularization_eps=opt_regularization_eps;
    }

#endif //HAVE_LAPACK

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

private:
    void init();

public:
#ifdef HAVE_LAPACK

    static const float64_t* get_Q_col(uint32_t i);

    static void print_state(libqp_state_T state);
#endif //HAVE_LAPACK

protected:
    index_t m_opt_max_iterations;

    float64_t m_opt_epsilon;

    float64_t m_opt_low_cut;

    float64_t m_opt_regularization_eps;

    static SGMatrix<float64_t> m_Q;
};

}

#endif /* __LINEARTIMEMMD_H_ */