Блог

Мне действительно нужна ваша помощь! Я написал этот код:

 from sklearn.model_selection import train_test_split
from sklearn import metrics
from sklearn.metrics import accuracy_score

def train_test_rmse(x,y):
    X = df_new[feature_cols]
    y = df_new['TOTAL CONSTRUCTION COST - EXCLUDING TAX']
    X_train, X_test, y_train, y_test = train_test_split(x, y, test_size = 0.2,random_state=123)
    linreg = LinearRegression()
    linreg.fit(X_train, y_train)
    y_test = train_test_split(x, y, test_size = 0.2,random_state=123)
    y_pred = linreg.predict(X_test)
    print(accuracy_score(y_test, y_pred))  
    return np.sqrt(metrics.mean_squared_error(y_test, y_pred)) 
 

^ Приведенный выше код выполняется правильно. Но когда я пытаюсь построить точечный график в ячейке ниже:

 import matplotlib.pyplot as plt
plt.scatter(y_test, y_pred)
plt.xlabel('Actual Y')
plt.ylabel('Predicted Y')
plt.show()
 

Я получаю сообщение об ошибке «имя’y_test’ не определено». Пожалуйста, дайте мне знать, как это исправить. Спасибо.

В коде я вижу, что y_test это определено внутри train_test_rmse функции, вам нужно инициализировать y_test вне этой функции.

ваш код должен работать нормально с небольшими изменениями следующим образом :

 y_test = None

def train_test_rmse(x,y):
  
  global y_test
  
    X = df_new[feature_cols]
    y = df_new['TOTAL CONSTRUCTION COST - EXCLUDING TAX']
    X_train, X_test, y_train, y_test = train_test_split(x, y, test_size = 0.2,random_state=123)