Fastai中的Logloss指标

3
我正在参加Zindi平台上的比赛,该比赛使用Log Loss作为评估指标。我使用fastai库进行开发,需要计算Log Loss,但是在这个库中没有找到LogLoss指标! 我尝试了一些代码,如从sklearn导入提供的函数from sklearn.metrics import log_loss,但是它没有起作用。 比赛链接: https://zindi.africa/competitions/basic-needs-basic-rights-kenya-tech4mentalhealth

我不了解FastAI,但PyTorch有负对数似然损失nn.NLLLossnn.CrossEntropyLoss,具体取决于您是否已经有对数概率。 - Michael Jungo
1个回答

2
如果需要作为度量标准(通常主要用作损失),您可以使用pytorch中的交叉熵函数进行计算:

import torch.nn.functional as F

metrics=[F.cross_entropy,(如果需要其他度量标准,请添加)]

model= cnn_learner(data, model, metrics=metrics,...)

请记得保留html标签,谢谢!

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接