【发布时间】:2017-04-04 21:21:26
【问题描述】:
我想在一个数据科学项目中进行预测,误差是通过非对称函数计算出来的。
是否可以调整随机森林或梯度提升(sklearn)的损失函数?
我已阅读需要修改 .pyx 文件,但我在我的 sklearn 文件夹中找不到任何文件(我在 ubuntu 14.04 LTS 上)。
你有什么建议吗?
【问题讨论】:
标签: python scikit-learn
我想在一个数据科学项目中进行预测,误差是通过非对称函数计算出来的。
是否可以调整随机森林或梯度提升(sklearn)的损失函数?
我已阅读需要修改 .pyx 文件,但我在我的 sklearn 文件夹中找不到任何文件(我在 ubuntu 14.04 LTS 上)。
你有什么建议吗?
【问题讨论】:
标签: python scikit-learn
是的,可以调整。例如:
class ExponentialPairwiseLoss(object):
def __init__(self, groups):
self.groups = groups
def __call__(self, preds, dtrain):
labels = dtrain.get_label().astype(np.int)
rk = len(np.bincount(labels))
plus_exp = np.exp(preds)
minus_exp = np.exp(-preds)
grad = np.zeros(preds.shape)
hess = np.zeros(preds.shape)
pos = 0
for size in self.groups:
sum_plus_exp = np.zeros((rk,))
sum_minus_exp = np.zeros((rk,))
for i in range(pos, pos + size, 1):
sum_plus_exp[labels[i]] += plus_exp[i]
sum_minus_exp[labels[i]] += minus_exp[i]
for i in range(pos, pos + size, 1):
grad[i] = -minus_exp[i] * np.sum(sum_plus_exp[:labels[i]]) +\
plus_exp[i] * np.sum(sum_minus_exp[labels[i] + 1:])
hess[i] = minus_exp[i] * np.sum(sum_plus_exp[:labels[i]]) +\
plus_exp[i] * np.sum(sum_minus_exp[labels[i] + 1:])
pos += size
return grad, hess
【讨论】:
【讨论】: