【发布时间】:2019-06-13 00:47:08
【问题描述】:
如何在 scikit learn 中自定义损失函数?例如,我想使用 MSE 乘以样本的真实值,而不是使用均方误差。我使用了以下代码sn-p:
def my_custom_loss_func(y_true,y_pred):
diff3=(abs(y_true-y_pred))*y_true
return diff3
clf=RandomForestRegressor(criterion=my_custom_loss_func)
knn=clf.fit(feam,labm)
我收到以下错误:
KeyError: <function my_custom_loss_func at 0x000000002EA9CA60>
【问题讨论】:
-
@franco piccolo 传递给 my_custom_loss_func 的参数应该是什么? scikit learn 中是否预定义了 y_pred 和 y_true?我的标签矩阵是labm。如果我用labm代替y_true,我应该用什么代替y_pred?
标签: python python-3.x machine-learning scikit-learn loss-function