WeightedDegreeRBFKernel.h

Go to the documentation of this file.

#ifndef _WEIGHTEDDEGREERBFKERNEL_H___
#define _WEIGHTEDDEGREERBFKERNEL_H___

#include "lib/common.h"
#include "kernel/SimpleKernel.h"
#include "features/SimpleFeatures.h"

namespace shogun
{

class CWeightedDegreeRBFKernel: public CSimpleKernel<float64_t>
{
    public:
        CWeightedDegreeRBFKernel();

        CWeightedDegreeRBFKernel(int32_t size, float64_t width, int32_t degree, int32_t nof_properties);

        CWeightedDegreeRBFKernel(CSimpleFeatures<float64_t>* l, CSimpleFeatures<float64_t>* r,
            float64_t width, int32_t degree, int32_t nof_properties, int32_t size=10);

        virtual ~CWeightedDegreeRBFKernel();

        virtual bool init(CFeatures* l, CFeatures* r);

        //virtual EKernelType get_kernel_type() { return K_UNKNOWN; }
        virtual EKernelType get_kernel_type() { return K_WEIGHTEDDEGREERBF; }
        
        inline virtual const char* get_name() const { return "WeightedDegreeRBFKernel"; }
        
    
    
        inline bool set_degree(int32_t deg) { degree=deg; return true; }

        inline int32_t get_degree() { return degree; }

        protected:
        virtual float64_t compute(int32_t idx_a, int32_t idx_b);

        bool init_wd_weights();


#ifdef HAVE_BOOST_SERIALIZATION
    private:

        friend class ::boost::serialization::access;
        template<class Archive>
            void serialize(Archive & ar, const unsigned int archive_version)
            {

                SG_DEBUG("archiving WeightedDegreeRBFKernel\n");

                ar & ::boost::serialization::base_object<CSimpleKernel<float64_t> >(*this);

                ar & width;

                SG_DEBUG("done with WeightedDegreeRBFKernel\n");

            }
#endif //HAVE_BOOST_SERIALIZATION


    protected:
        float64_t width;

        int32_t degree;

        int32_t nof_properties;

        float64_t* weights;

};
}
#endif /* _WEIGHTEDDEGREERBFKERNEL_H__ */