Вот несколько методов, предоставляемых модулем дерева решений scikit-learn:
- DecisionTreeClassifier: этот класс реализует алгоритм дерева решений для задач классификации.
- DecisionTreeRegressor: этот класс реализует алгоритм дерева решений для задач регрессии.
- export_graphviz: эта функция экспортирует дерево решений в формате Graphviz, которое можно визуализировать с помощью внешних инструментов.
- DecisionTreeClassifier.featureimportances: этот атрибут отражает важность каждой функции в модели дерева решений.
- DecisionTreeClassifier.predict: этот метод прогнозирует метки классов для заданного набора входных выборок.
- DecisionTreeRegressor.predict: этот метод прогнозирует целевые значения для заданного набора входных выборок.