cogdl.tasks.unsupervised_graph_classification

Module Contents

Classes

UnsupervisedGraphClassification

Unsupervised graph classification
class cogdl.tasks.unsupervised_graph_classification.UnsupervisedGraphClassification(args, dataset=None, model=None)[source]

Bases: cogdl.tasks.BaseTask

Unsupervised graph classification

static add_args(parser)[source]

Add task-specific arguments to the parser.

train(self)[source]
save_emb(self, embs)[source]
_evaluate(self, embeddings, labels)[source]