GaussianNaiveBayes.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) 2011 Sergey Lisitsyn
 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
 */

#ifndef GAUSSIANNAIVEBAYES_H_
#define GAUSSIANNAIVEBAYES_H_

#include <shogun/lib/config.h>

#include <shogun/machine/NativeMulticlassMachine.h>
#include <shogun/mathematics/Math.h>
#include <shogun/features/DotFeatures.h>

namespace shogun {

class CLabels;
class CDotFeatures;
class CFeatures;

class CGaussianNaiveBayes : public CNativeMulticlassMachine
{

public:
    MACHINE_PROBLEM_TYPE(PT_MULTICLASS)

    
    CGaussianNaiveBayes();

    CGaussianNaiveBayes(CFeatures* train_examples, CLabels* train_labels);

    virtual ~CGaussianNaiveBayes();

    virtual void set_features(CFeatures* features);

    virtual CFeatures* get_features();

    virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);

    virtual float64_t apply_one(int32_t idx);

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

    virtual EMachineType get_classifier_type() { return CT_GAUSSIANNAIVEBAYES; };

protected:

    virtual bool train_machine(CFeatures* data=NULL);

protected:

    CDotFeatures* m_features;

    int32_t m_min_label;

    int32_t m_num_classes;

    int32_t m_dim;

    SGMatrix<float64_t> m_means;

    SGMatrix<float64_t> m_variances;

    SGVector<float64_t> m_label_prob;

    SGVector<float64_t> m_rates;
};

}

#endif /* GAUSSIANNAIVEBAYES_H_ */