BALL 1.5.0
knnModel.h
Go to the documentation of this file.
1// -*- Mode: C++; tab-width: 2; -*-
2// vi: set ts=2:
3//
4//
5
6#ifndef BALL_QSAR_KNNMODEL_H
7#define BALL_QSAR_KNNMODEL_H
8
9#ifndef BALL_QSAR_ALLMODEL_H
10#include <BALL/QSAR/allModel.h>
11#endif
12
13namespace BALL
14{
15 namespace QSAR
16 {
17
20 {
21 public:
25 KNNModel(const QSARData& q, int k=3);
27
28 void calculateWeights(Eigen::MatrixXd& dist, Eigen::VectorXd& w);
29
30 void setParameters(vector<double>& v);
31
32 vector<double> getParameters() const;
33
34 bool optimizeParameters(int k, int no_steps);
35
36 private:
37 int k_ ;
38 };
39 }
40}
41
42#endif // KNNMODEL
Definition: constants.h:13
BALL_EXTERN_VARIABLE const double k
Definition: constants.h:93
void setParameters(vector< double > &v)
vector< double > getParameters() const
KNNModel(const QSARData &q, int k=3)
bool optimizeParameters(int k, int no_steps)
void calculateWeights(Eigen::MatrixXd &dist, Eigen::VectorXd &w)
#define BALL_EXPORT
Definition: COMMON/global.h:50