lgd_gen_proof_code/DataSet.py at main · apd10/lgd_gen_proof_code · 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
#from dataparsers.list_dataparsers import GenSVMFormatParser, CSVParser, RaceSampler, RaceSamplerPreProc, RaceGenSamplerPreProc, RunningRaceParser
from dataparsers.list_dataparsers import *


class DataSet:
    ''' specifies the parsing for the data
    '''
    def get(data_file, name, params):
        if name == "gensvm":
          dataset = GenSVMFormatParser(data_file, params)
        elif name == "csv":
          dataset = CSVParser(data_file, params)
        elif name == "csv_1hnn":
          dataset = CSV1hnnParser(data_file, params)
        elif name == "bin":
          dataset = BinParser(data_file, params)
        elif name == "race":
          dataset = RaceSampler(data_file, params)
        elif name == "race_pp":
          dataset = RaceSamplerPreProc(data_file, params)
        elif name == "race_ppd":
          dataset = RaceSamplePPD(data_file, params)
        elif name == "race_hist":
          dataset = RaceSampleHist(data_file, params)
        elif name == "race_gen_pp":
          dataset = RaceGenSamplerPreProc(data_file, params)
        elif name == "running_race":
          dataset = RunningRaceParser(data_file, params)
        elif name == "gaussian_kde":
          dataset = GaussianKDE(data_file, params)
        else:
          raise NotImplementedError
        return dataset