Итерации для многочленов разных порядков R-квадрат с циклом for; IndexError: только целые числа, кусочки (`:`), многоточие (`...`), numpy - PullRequest
Новая
       3

Итерации для многочленов разных порядков R-квадрат с циклом for; IndexError: только целые числа, кусочки (`:`), многоточие (`...`), numpy

0 голосов
/ 26 марта 2020

Я получаю эту ошибку в 7-й строке (x_train_pr = pr.fit_transform(x_train[['horsepower']])) кода при создании «для l oop», чтобы увидеть, как R ^ 2 изменяется в моих тестовых данных для полиномов различного порядка, и построить результаты.

Я столкнулся с похожими проблемами, но решил их, добавив .values к подмножеству (например, df[['horsepower']].values) или reshape(-1,1), или используя одну квадратную скобку, но они здесь не работают. Я также убедился, что тип данных для horsepower равен int. Наконец, я попытался применить код к другому набору данных. по некоторым причинам это работает только на когнитивной лаборатории класса. Я новичок в программировании и изучении науки о данных на cognitiveclass.ai. Таким образом, набор данных, который я использую, является автоматическим набором данных, рекомендованным для класса.

Вот код 

Rsqu_test = []

order = [1, 2, 3, 4]
for n in order:
    pr = PolynomialFeatures(degree=n)

    x_train_pr = pr.fit_transform(x_train[['horsepower']])

    x_test_pr = pr.fit_transform(x_test[['horsepower']])    

    lr.fit(x_train_pr, y_train)

    Rsqu_test.append(lr.score(x_test_pr, y_test))
...