WeightedDegreeRBFKernel.h

Go to the documentation of this file.

#ifndef _WEIGHTEDDEGREERBFKERNEL_H___
#define _WEIGHTEDDEGREERBFKERNEL_H___

#include <shogun/lib/config.h>

#include <shogun/lib/common.h>
#include <shogun/kernel/DotKernel.h>
#include <shogun/features/DenseFeatures.h>

namespace shogun
{

class CWeightedDegreeRBFKernel: public CDotKernel
{
    public:
        CWeightedDegreeRBFKernel();

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

        CWeightedDegreeRBFKernel(CDenseFeatures<float64_t>* l, CDenseFeatures<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; }

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


        virtual EFeatureClass get_feature_class() { return C_DENSE; }

        virtual EFeatureType get_feature_type() { return F_DREAL; }


        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();

    protected:
        float64_t width;

        int32_t degree;

        int32_t nof_properties;

        float64_t* weights;

    private:
        void register_params();

};
}
#endif /* _WEIGHTEDDEGREERBFKERNEL_H__ */