def __init__(self, symb, predlen, cat='RL', kwargs=None): self.symb = symb self.predlen = predlen self.kwargs = kwargs self.cat = cat if cat == 'RF': if kwargs != None: self.learner = RF.RandomForest(**kwargs) else: self.learner = RF.RandomForest() elif cat == 'KNN': if kwargs != None: self.learner = KNN.KNN(**kwargs) else: self.learner = KNN.KNN() elif cat == 'SVM': if kwargs != None: self.learner = SVM.SVM(**kwargs) else: self.learner = SVM.SVM() elif cat == 'NN': if kwargs != None: self.learner = NN.NN(**kwargs) else: self.learner = NN.NN()