tasks
¶
Submodules¶
tasks.base_task
tasks.graph_classification
tasks.heterogeneous_node_classification
tasks.link_prediction
tasks.multiplex_link_prediction
tasks.multiplex_node_classification
tasks.node_classification
tasks.node_classification_sampling
tasks.unsupervised_graph_classification
tasks.unsupervised_node_classification
Package Contents¶
Functions¶
|
New task types can be added to cogdl with the |
|
-
class
tasks.
BaseTask
(args)[source]¶ Bases:
object
-
static
add_args
(parser)¶ Add task-specific arguments to the parser.
-
abstract
train
(self, num_epoch)¶
-
static
-
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