Машинное обучение - со склеарном - PullRequest
0 голосов
/ 20 января 2019

Я пишу кусок кода для базового прогноза акций, но получаю следующую ошибку.

AttributeError: у объекта 'function' нет атрибута 'train_test_split'

Мой код, кроме всего этого, кажется правильным и был запущен для тестирования в процессе кодирования. Так что я уверен, что нет другой проблемы, вызывающей это, кроме проблемы с библиотекой Python. Кто-нибудь знает решение этой проблемы, чтобы проект мог продолжаться? Это мой код, если это поможет.

import quandl, math
import numpy as np
import pandas as pd
from sklearn import preprocessing, svm
from sklearn.model_selection import cross_validate
from sklearn.linear_model import LinearRegression

#Getting the data
df = quandl.get("WIKI/GOOGL")

#Selecting the data we want from the database
df = df[['Adj. Open','Adj. High','Adj. Low','Adj. Close','Adj. Volume']]

#Calculating percentage changes
df['HL_PCT'] = (df['Adj. High'] - df['Adj. Close']) / df['Adj. Close'] * 100
df['PCT_change'] = (df['Adj. Close'] - df['Adj. Open']) / df['Adj. Open'] * 100

#Refining the data even further
df = df[['Adj. Close', 'HL_PCT', 'PCT_change', 'Adj. Volume']] 

forecast_col = 'Adj. Close'
df.fillna(value=-99999, inplace=True)
forecast_out = int(math.ceil(0.01 * len(df)))

df['label'] = df[forecast_col].shift(-forecast_out)


x = np.array(df.drop(['label'],1))
y = np.array(df['label'])
x  = preprocessing.scale(x)
y = np.array(df['label'])

X_train, X_test, y_train, y_test = cross_validate.train_test_split(X, y, test_size=0.2)

1 Ответ

0 голосов
/ 20 января 2019

Проблема в train_test_split не внутри sklearn.crossvalidate, а внутри sklearn.model_selection.Если вы хотите использовать train_test_split, вы должны использовать его как-

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

Для получения более подробной информации просмотрите этот URL - https://scikit -learn.org / stable / modules / generate / sklearn.model_selection.train_test_split.html

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