Полиномиальная регрессия плохой сюжет - PullRequest
0 голосов
/ 02 апреля 2020

Я новичок во всем этом, поэтому я много искал решение и читал блоги и т.д. c. но безрезультатно, если бы вы могли объяснить проблему вместе с решением, я был бы очень признателен.

[! [] [1]] [1]

код:


import pandas as pd

df = pead_csv(

x = StandardScaler().fit_transform(df)



}

d
    return
viz_polymonial()

1 Ответ

0 голосов
/ 02 апреля 2020

Во-первых, вы выполняете подгонку полиномиальных функций дважды, это ваше предназначение?

Описанная проблема возникает из-за того, что plt.plot(X, Y) соединяется со строками (X[0], Y[0]) с (X[1], Y[1]), (X[1], Y[1]) с (X[2], Y[2]), et c. Порядок имеет значение, если вы перетасуете X и Y, вы получите другой сюжет. Здесь ваши точки неупорядочены, вам нужно отсортировать их, чтобы получить более плавный график.

def viz_polymonial():
    plt.scatter(X.iloc[:, 1], Y, color='red')
    data = list(zip(X, pol_reg.predict(poly_reg.transform(X)))
    # sort data by X and unpack
    sorted_X, sorted_Y = **sorted(data)
    plt.plot(sorted_X, sorted_Y, color='blue')
    plt.xlabel('Age. Annual income')
    plt.ylabel('Spending Score')
    plt.show()
    return
viz_polymonial()

Если вы сделаете минимальный воспроизводимый пример, я протестирую его на своей машине.

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