Mglearn сюжеты с неправильными цветами - PullRequest
0 голосов
/ 09 октября 2019

Итак, прежде всего, я должен сказать, что я новичок в Python и так далее ... поэтому извините меня заранее, если мой вопрос глупый!

Я сейчас пытаюсь научитьсякое-что о машинном обучении и, в частности, об использовании mglearn. Я хотел сделать Регрессионную классификацию и для этого я использовал Linear SVM.

Я пробовал этот скрипт:

import mglearn
import matplotlib.pyplot as plt
import numpy as np
from sklearn.datasets import make_blobs
X, y = make_blobs(random_state=42)
from sklearn.svm import LinearSVC
linear_svm = LinearSVC().fit(X, y)
mglearn.plots.plot_2d_classification(linear_svm, X, fill=True, alpha=.7)

mglearn.discrete_scatter(X[:, 0], X[:, 1], y)
line = np.linspace(-15, 15)
for coef, intercept, color in zip(linear_svm.coef_, linear_svm.intercept_, ['b', 'r', 'g']):
    plt.plot(line, -(line * coef[0] + intercept) / coef[1])

plt.ylim(-10, 15)
plt.xlim(-10, 8)
plt.xlabel("Feature 0")
plt.ylabel("Feature 1")
plt.legend(['Class 0', 'Class 1', 'Class 2', 'Line class 0', 'Line class 1', 'Line class 2'], loc=(1.01, 0.3))

Моя проблема в том, что когда я делаю это, в соответствии спрочитайте, точки и «фон» точек должны быть одного цвета. И как вы можете видеть на картинке, у меня есть ссылка, зеленые точки на красном фоне и красная точка на зеленом фоне.

Есть идеи?

Сюжет с неправильными цветами ...

Спасибо людям:)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...