Задача классификации

Обучение с учителем с помощью scikit-learn

George Boorman

Core Curriculum Manager, DataCamp

Классификация неразмеченных данных

  1. Создаём модель
  2. Модель обучается на размеченных данных
  3. Передаём неразмеченные данные на вход модели
  4. Модель предсказывает метки новых данных

 

  • Размеченные данные = обучающие данные
Обучение с учителем с помощью scikit-learn

k ближайших соседей

  • Метка точки данных предсказывается

    • На основе k ближайших размеченных точек

    • Путём голосования большинством

Обучение с учителем с помощью scikit-learn

k ближайших соседей

точечный график с наблюдениями синего и красного цвета и новым наблюдением чёрного цвета

Обучение с учителем с помощью scikit-learn

k ближайших соседей

радиус вокруг трёх наблюдений, ближайших к чёрной точке

Обучение с учителем с помощью scikit-learn

k ближайших соседей

радиус вокруг пяти наблюдений, ближайших к чёрной точке

Обучение с учителем с помощью scikit-learn

Интуиция KNN

точечный график зависимости вечерних расходов от дневных: наблюдения окрашены в синий цвет при оттоке клиента и в красный — при его отсутствии

Обучение с учителем с помощью scikit-learn

Интуиция KNN

точечный график оттока с границей решения, разделяющей наблюдения по прогнозу KNN

Обучение с учителем с помощью scikit-learn

Обучение классификатора с помощью scikit-learn

from sklearn.neighbors import KNeighborsClassifier

X = churn_df[["total_day_charge", "total_eve_charge"]].values
y = churn_df["churn"].values

print(X.shape, y.shape)

(3333, 2), (3333,)

knn = KNeighborsClassifier(n_neighbors=15)

knn.fit(X, y)
Обучение с учителем с помощью scikit-learn

Предсказание на неразмеченных данных

X_new = np.array([[56.8, 17.5],
                  [24.4, 24.1],
                  [50.1, 10.9]])

print(X_new.shape)

(3, 2)

predictions = knn.predict(X_new)

print('Predictions: {}'.format(predictions))

Predictions: [1 0 0]
Обучение с учителем с помощью scikit-learn

Давайте потренируемся!

Обучение с учителем с помощью scikit-learn

Preparing Video For Download...