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