Как исправить ошибку значения с train_test_split в Python Numpy - PullRequest
0 голосов
/ 31 мая 2019

Я использую sklearn с массивом numpy.У меня есть 2 массива (x, y), и они должны быть:

test_size=0.2
train_size=0.8

Это мой текущий код:

def predict():

    sample_data = pd.read_csv("includes\\csv.csv")

    x = np.array(sample_data["day"])
    y = np.array(sample_data["balance"])


    x = x.reshape(1, -1)



    y = y.reshape(1, -1)




    print(x)
    print(y)



    X_train, X_test, y_train, y_test = train_test_split(x, y, test_size=0.2)



    clf = LinearRegression()
    clf.fit(x_train, y_train)

    clf.score(x_test, y_test)

Ошибка:

ValueError: With n_samples=1, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.

и оно появляется в строке:

X_train, X_test, y_train, y_test = train_test_split(x, y, test_size=0.2)

Есть идеи, почему это появляется?

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