shenweichen / DeepCTR

Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
https://deepctr-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
7.54k stars 2.21k forks source link

ple可以只用于单任务吗 #525

Open leiqing110 opened 1 year ago

leiqing110 commented 1 year ago

作者您好,看到源码中限制了任务的数量,必须要大于1,可以修改这里的源码,用于单任务吗 ple源码 ` num_tasks = len(task_names) if num_tasks <= 1: raise ValueError("num_tasks must be greater than 1")

if len(task_types) != num_tasks:
    raise ValueError("num_tasks must be equal to the length of task_types")

for task_type in task_types:
    if task_type not in ['binary', 'regression']:
        raise ValueError("task must be binary or regression, {} is illegal".format(task_type))

`