def __init__(self, method='', dir_name='temp', is_clean=True):
self.prefix = 'texts/'
self.arff = Parse_ARFF()
self.dir_name = dir_name
if is_clean: self.__clean_dir(self.prefix + self.dir_name)
self.n_folds = 5
self.classes = [0, 1]
self.method_prev = self._bin_prevalence #._bin_prevalence or ._multi_prevalence
self.model = self.__classificator(class_weight='auto')
if method == 'EM' or method == 'EM1' or method == 'Iter' or method == 'Iter1':
self.method = method
elif method == 'PCC' or method == 'CC' or method == 'ACC' or method == 'PACC':
self.method = method
elif method == 'test':
self.method = method
self._train_file, self._test_files = self.arff.read_dir(
self.prefix + 'pickle_' + dir_name)
elif method == '':
self.method = 'CC'
