shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.
https://deepctr-torch.readthedocs.io/en/latest/index.html
Apache License 2.0
2.95k stars 696 forks source link

MMOE metric的代码有些问题,不适合分类+回归的任务 #287

Open Shuigs18 opened 1 year ago

Shuigs18 commented 1 year ago

Describe the bug(问题描述) 任务为二分类和回归时,评价指标必须设置成二者都能使用的评价函数

BaseModel 模块中

if verbose > 0: for name, metric_fun in self.metrics.items(): if name not in train_result: train_result[name] = [] train_result[name].append(metric_fun( y.cpu().data.numpy(), y_pred.cpu().data.numpy().astype("float64"))) 这里的y是每个任务输出的拼接,是不是可以改成不同的任务不同的评价函数