MPDSVM.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) 1999-2009 Soeren Sonnenburg
 * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */

#ifndef _MPDSVM_H___
#define _MPDSVM_H___
#include <shogun/lib/common.h>
#include <shogun/classifier/svm/SVM.h>
#include <shogun/lib/Cache.h>

namespace shogun
{
class CMPDSVM : public CSVM
{
    public:
        CMPDSVM();

        CMPDSVM(float64_t C, CKernel* k, CLabels* lab);
        virtual ~CMPDSVM();

        virtual inline EClassifierType get_classifier_type() { return CT_MPD; }

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

    protected:
        virtual bool train_machine(CFeatures* data=NULL);

        inline float64_t compute_H(int32_t i, int32_t j)
        {
            return labels->get_label(i)*labels->get_label(j)*kernel->kernel(i,j);
        }

        inline KERNELCACHE_ELEM* lock_kernel_row(int32_t i)
        {
            KERNELCACHE_ELEM* line=NULL;

            if (kernel_cache->is_cached(i))
            {
                line=kernel_cache->lock_entry(i);
                ASSERT(line);
            }

            if (!line)
            {
                line=kernel_cache->set_entry(i);
                ASSERT(line);

                for (int32_t j=0; j<labels->get_num_labels(); j++)
                    line[j]=(KERNELCACHE_ELEM) labels->get_label(i)*labels->get_label(j)*kernel->kernel(i,j);
            }

            return line;
        }

        inline void unlock_kernel_row(int32_t i)
        {
            kernel_cache->unlock_entry(i);
        }

        CCache<KERNELCACHE_ELEM>* kernel_cache;
};
}
#endif  /* _MPDSVM_H___ */