Я новичок в машинном обучении и получаю эту ошибку, когда пытаюсь выполнить метод .predict()
на X_test
. Я также не понимаю концепцию reshape
: зачем нам это делать?
# Importing the libraries
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# Importing the dataset
dataset = pd.read_csv('Salary_Data.csv')
X = dataset.iloc[:,-1].values
y = dataset.iloc[:,1].values
# Splitting the dataset into the Training set and Test set
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 1/3, random_state = 0)
# Fitting simple Linear Regression to the training set
from sklearn.linear_model import LinearRegression
regressor= LinearRegression()
X_train = X_train.reshape(1, -1)
y_train = y_train.reshape(1, -1)
regressor.fit(X_train,y_train)
# Predicting the test set results
X_test= X_test.reshape(1,-1)
y_pred = regressor.predict(X_test)