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