Esempio n. 1
0
    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