BALL  1.5.0
ldaModel.h
Go to the documentation of this file.
1 // -*- Mode: C++; tab-width: 2; -*-
2 // vi: set ts=2:
3 //
4 //
5 
6 #ifndef LDAMODEL
7 #define LDAMODEL
8 
9 #ifndef CLASSIFICATION
11 #endif
12 
13 #ifndef STATISTICS
14 #include <BALL/QSAR/statistics.h>
15 #endif
16 
17 namespace BALL{
18  namespace QSAR {
19 
22  {
23  public:
27  LDAModel(const QSARData& q);
28 
31 
32 
36  void train();
37 
38  Eigen::VectorXd predict(const vector<double>& substance, bool transform=1);
39 
40  void setParameters(vector<double>& v);
41 
42  vector<double> getParameters() const;
43 
44  void saveToFile(string filename);
45 
46  void readFromFile(string filename);
48 
49 
50  private:
55  Eigen::MatrixXd sigma_;
56 
57 
58  double lambda_;
59 
61  vector<Eigen::MatrixXd > mean_vectors_;
63  };
64  }
65 }
66 
67 #endif //LDAMODEL
BALL_EXPORT
#define BALL_EXPORT
Definition: COMMON/global.h:50
BALL::QSAR::LDAModel::saveToFile
void saveToFile(string filename)
BALL::QSAR::LDAModel::LDAModel
LDAModel(const QSARData &q)
BALL::QSAR::ClassificationModel
Definition: classificationModel.h:28
BALL::QSAR::LDAModel::~LDAModel
~LDAModel()
BALL::QSAR::LDAModel::train
void train()
BALL
Definition: constants.h:13
BALL::QSAR::LDAModel::setParameters
void setParameters(vector< double > &v)
BALL::QSAR::QSARData
Definition: QSARData.h:56
BALL::QSAR::LDAModel::readFromFile
void readFromFile(string filename)
statistics.h
BALL::QSAR::LDAModel
Definition: ldaModel.h:22
classificationModel.h
BALL::QSAR::LDAModel::predict
Eigen::VectorXd predict(const vector< double > &substance, bool transform=1)
BALL::QSAR::LDAModel::getParameters
vector< double > getParameters() const