1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
#include <BALLTestConfig.h>
#include <BALL/CONCEPT/classTest.h>
#include <BALL/QSAR/QSARData.h>
#include <BALL/QSAR/mlrModel.h>
using namespace BALL;
using namespace BALL::QSAR;
START_TEST(MLR-model)
PRECISION(1E-7)
QSARData data;
data.readCSVFile(BALL_TEST_DATA_PATH(Regression_test.csv),1,1,1," ",0,0);
CHECK(MLR-model)
MLRModel mlr(data);
mlr.readTrainingData();
mlr.train();
const Eigen::MatrixXd* res = mlr.getTrainingResult();
TEST_EQUAL(res->rows(),3)
TEST_REAL_EQUAL((*res)(0,0),0.1407737)
TEST_REAL_EQUAL((*res)(1,0),0.5902179)
TEST_REAL_EQUAL((*res)(2,0),0.4159182)
RESULT
END_TEST
|