StringKernel4LIBSVM/StringKernelConverter.cpp at master · cttsai1985/StringKernel4LIBSVM · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
#include "GetLongOpt.h"
#include "StringKernels.h"
#include <cstdio>
#include <string>

using namespace std;

//int buffer_size=65536;

//setting
int kernel_type=0;
int kmer;

int main(int argc, char** argv)
{
	int error_var=0;
	GetLongOpt GLO(argc, argv);
	kernel_type=GLO.GetIntOpt("t", 1);
	kmer=GLO.GetIntOpt("k", 5);


	string train_filename=GLO.GetStringOpt("train", "");
	string test_filename=GLO.GetStringOpt("test", "");

	KernelConfig kernel_config(kernel_type, kmer);
	kernel_config.SetCoef();

	DataSet* target_dataset;
	DataSet train_dataset, test_dataset;
	train_dataset.initialize();
	test_dataset.initialize();

	error_var=LoadDataset(train_filename, &train_dataset);
	if(error_var==-1)
		return 0;
	if(test_filename.empty()==false)
	{
		error_var=LoadDataset(test_filename, &test_dataset);
		target_dataset=&test_dataset;
		if(error_var==-1)
			return 0;
	}
	else
		target_dataset=&train_dataset;

	//PreComputeWDKernel(&kernel_config, &train_dataset, target_dataset);
	//WeightingKernelOuptputAB(&kernel_config, target_dataset);

	switch(kernel_config.kernel_type)
	{
		case  DSK:
			ComputeDKernel(&kernel_config, &train_dataset, target_dataset);
			break;
		case WDSK:
			ComputeWDKernel(&kernel_config, &train_dataset, target_dataset);
			break;
	}
    return 0;
}