Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
Shuigs18 opened this issue Jul 20, 2023 · 0 comments
Open

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

Shuigs18 opened this issue Jul 20, 2023 · 0 comments
Labels
question Further information is requested

Comments

@Shuigs18
Copy link

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是每个任务输出的拼接,是不是可以改成不同的任务不同的评价函数

@Shuigs18 Shuigs18 added the question Further information is requested label Jul 20, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

1 participant