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
|
#include <BALLTestConfig.h>
#include <BALL/CONCEPT/classTest.h>
#include <BALL/QSAR/QSARData.h>
#include <BALL/QSAR/nBModel.h>
using namespace BALL;
using namespace BALL::QSAR;
START_TEST(NB-model)
PRECISION(1E-7)
QSARData data;
data.readCSVFile(BALL_TEST_DATA_PATH(Regression_test.csv),1,1,1," ",0,0);
CHECK(NB-model)
NBModel model(data);
model.readTrainingData();
model.train();
model.validation->testInputData();
double quality=model.validation->getFitRes();
TEST_REAL_EQUAL(quality,1.0)
RESULT
END_TEST
|