cogdl.tasks

Package Contents

Classes

BaseTask

Functions

register_task(name)

New task types can be added to cogdl with the register_task()

build_task(args, dataset=None, model=None)

class cogdl.tasks.BaseTask(args)[source]

Bases: object

static add_args(parser)

Add task-specific arguments to the parser.

abstract train(self, num_epoch)
cogdl.tasks.TASK_REGISTRY[source]
cogdl.tasks.register_task(name)[source]

New task types can be added to cogdl with the register_task() function decorator.

For example:

@register_task('node_classification')
class NodeClassification(BaseTask):
    (...)
Args:

name (str): the name of the task

cogdl.tasks.task_name[source]
cogdl.tasks.build_task(args, dataset=None, model=None)[source]