bayesModel.h

Go to the documentation of this file.

// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
//
 
#ifndef BALL_QSAR_BAYESMODEL_H
#define BALL_QSAR_BAYESMODEL_H
 
#ifndef BALL_QSAR_CLASSIFICATIONMODEL_H
#include <BALL/QSAR/classificationModel.h>
#endif
 
namespace BALL
{
  namespace QSAR
  {
    class BALL_EXPORT BayesModel : public ClassificationModel
    {
      public:
 
        BayesModel(const QSARData& q);
 
        virtual bool isTrained() = 0;

        virtual vector<double> calculateProbabilities(int activitiy_index, int feature_index, double feature_value) = 0;

        virtual int getNoResponseVariables() = 0;
 
 
      protected:
        double min_prob_diff_;

        double undef_act_class_id_;
 
    };
  }
}
 
#endif //BAYESMODEL_H