def add_tree(self, name, begin, end, labels=None): print("new tree created") # Читаем базу проекта и выделяем пул и классы fixed_df = pd.read_csv(self.database) tree_pool, labels, translations = dt.bagging(fixed_df, end, begin, labels) # Создаём класс даты checkingData = dt.DataWorks(tree_pool, labels, translations) # Создаём и сохраняем дерево tree = dt.DecisionTree.createFromData( checkingData) # DecisionTree(checkingData) tree.drawTree() tree.saveTree(self.trees_dir + name + '.json') self.trees.append(self.trees_dir + name + '.json') self.save_project() # возвращаем список деревьев return self.trees