Loading...
Searching...
No Matches
ldaModel.h
Go to the documentation of this file.
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
//
#ifndef LDAMODEL
#define LDAMODEL
#ifndef CLASSIFICATION
#include <BALL/QSAR/classificationModel.h>
#endif
#ifndef STATISTICS
#include <BALL/QSAR/statistics.h>
#endif
namespace BALL{
namespace QSAR {
class BALL_EXPORT LDAModel : public ClassificationModel
{
public:
LDAModel(const QSARData& q);
~LDAModel();
void train();
Eigen::VectorXd predict(const vector<double>& substance, bool transform=1);
void setParameters(vector<double>& v);
vector<double> getParameters() const;
void saveToFile(string filename);
void readFromFile(string filename);
private:
Eigen::MatrixXd sigma_;
double lambda_;
vector<Eigen::MatrixXd > mean_vectors_;
};
}
}
#endif //LDAMODEL
BALL_EXPORT
#define BALL_EXPORT
Definition COMMON/global.h:50
BALL::QSAR::ClassificationModel::ClassificationModel
ClassificationModel(const QSARData &q)
BALL::QSAR::LDAModel::predict
Eigen::VectorXd predict(const vector< double > &substance, bool transform=1)
BALL::QSAR::LDAModel::setParameters
void setParameters(vector< double > &v)
BALL::QSAR::LDAModel::getParameters
vector< double > getParameters() const
BALL::QSAR::LDAModel::LDAModel
LDAModel(const QSARData &q)
BALL::QSAR::LDAModel::train
void train()
BALL::QSAR::LDAModel::~LDAModel
~LDAModel()
BALL::QSAR::LDAModel::saveToFile
void saveToFile(string filename)
BALL::QSAR::LDAModel::readFromFile
void readFromFile(string filename)
BALL::QSAR::QSARData
Definition QSARData.h:56
classificationModel.h
BALL::QSAR
Definition allModel.h:17
BALL
Definition constants.h:13
statistics.h