File: LDAModel_test.C

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 239,848 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; makefile: 93
file content (47 lines) | stat: -rw-r--r-- 1,292 bytes parent folder | download | duplicates (7)
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include <BALLTestConfig.h>
#include <BALL/CONCEPT/classTest.h>

#include <BALL/QSAR/QSARData.h>
#include <BALL/QSAR/ldaModel.h>

using namespace BALL;
using namespace BALL::QSAR;


START_TEST(LDA-model)

PRECISION(1E-7)


QSARData data;
data.readCSVFile(BALL_TEST_DATA_PATH(Regression_test.csv),1,1,1,"\t",0,0);

QSARData data2;
data2.readCSVFile(BALL_TEST_DATA_PATH(Regression_test2.csv),1,1,1,"\t",0,0);

CHECK(LDA-model)
	LDAModel model(data);
	model.readTrainingData();
	model.train();
	model.validation->testInputData();
	double quality=model.validation->getFitRes();
	TEST_REAL_EQUAL(quality, 1.0)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[0], 1.00)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[1], 1.00)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[2], 1.00)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[3], 1.00)
RESULT

CHECK(LDA-model2)
	LDAModel model(data2);
	model.readTrainingData();
	model.train();
	model.validation->testInputData();
	double quality=model.validation->getFitRes();
	TEST_REAL_EQUAL(quality, 0.75)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[0], 0.75)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[1], 1.00)
	TEST_REAL_EQUAL((*model.validation->getClassResults())[2], 0.50)
RESULT

END_TEST