sklearn

Вероятность классов

Функция sklearn predict_proba().

Подходит для решающего дерева,случайного леса, логистической регрессии.

count_probabilities = model.predict_proba(df)

Настройка глубины дерева max_depth

model = DecisionTreeClassifier(random_state=5678, max_depth=4)

Работа с scikit-learn

from sklearn.tree import DecisionTreeClassifier

model = DecisionTreeClassifier()

 

Библиотека scikit-learn, sklearn, описание

Библиотека scikit-learn

Возможности

  • препроцессинг, предобработка;
  • выбор модели;
  • регрессии;
  • классификации;
  • кластерный анализ;
  • уменьшение размерности.

Задачи подготовки данных к ML: стандартизация, нормализация данных, кодирование категориальных переменных.

Выбор модели

Возможности: перекрестная проверка, настройка гиперпараметров моделей с помощью поиска по сетке и др.

Подписка на RSS - sklearn