如何在 scikit-learn 中计算回归成本函数答案

【问题标题】：How compute cost function for regression in scikit-learn如何在 scikit-learn 中计算回归成本函数
【发布时间】：2021-02-07 18:43:15
【问题描述】：

我正在尝试进行线性回归，但不知道计算成本函数：

这是我的代码：

lr = LinearRegression()

lr.fit(X_train,y_train)  #X_train les caractéristiques et Y_train les données cibles 


#coef a1 et intercept a0

print(lr.coef_)
print(lr.intercept_)


###cost function 

def cout_fonction(X_train,y_train):
    
    m=len(y_train)
    return 1/(2*m)*np.sum((lr.fit(X_train,y_train)-y_train)**2)


print(cout_fonction(X_train,y_train))

####

错误：

  File "C:\Users\selai\Anaconda3\lib\site-packages\pandas\core\ops\__init__.py", line 450, in masked_arith_op
    assert is_scalar(y), type(y)

AssertionError: <class 'sklearn.linear_model.base.LinearRegression'>

【问题讨论】：

标签： python python-3.x scikit-learn linear-regression mean-square-error

【解决方案1】：

IIUC，您想获得mean_squared_error。只需导入它并计算它：

from sklearn.metrics import mean_squared_error
print("MSE on train {:.3f}".format(mean_squared_error(y_train, lr.predict(X_train))))

【讨论】：