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()
