我不断收到错误名称'y_test'未定义

Question

我真的需要你的帮助！我已经写了这段代码：

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'未定义”。请让我知道如何解决它。谢谢。

Answer 1

在代码中，我看到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)