python - 如何在 scikit-learn 中制作加权平均绝对误差评分器？

Question

我正在尝试解决回归任务，并且要了解模型的性能，我需要使用 WMAE。但该指标在 Scikit-Learn 中不可用。计算这个指标的公式是这样的——

def WMAE(dataset, real, predicted):
    weights = dataset.IsHoliday.apply(lambda x: 5 if x else 1)
    return np.round(np.sum(weights*abs(real-predicted))/(np.sum(weights)), 2)

但我不确定如何将此函数转换为 scikit-learn 记分器。我需要这个才能在 cross_val_score 和 RandomSearchCV 中使用记分器。

任何人都可以帮忙吗？

标签： pythonscikit-learn