Я использую 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)
Есть идеи, почему это появляется?