cogdl.tasks.unsupervised_node_classification

Module Contents

Classes

UnsupervisedNodeClassification

Node classification task.

TopKRanker

cogdl.tasks.unsupervised_node_classification.pyg = False[source]
class cogdl.tasks.unsupervised_node_classification.UnsupervisedNodeClassification(args, dataset=None, model=None)[source]

Bases: cogdl.tasks.BaseTask

Node classification task.

static add_args(parser)[source]

Add task-specific arguments to the parser.

enhance_emb(self, G, embs)[source]
save_emb(self, embs)[source]
train(self)[source]
_evaluate(self, features_matrix, label_matrix, num_shuffle)[source]
class cogdl.tasks.unsupervised_node_classification.TopKRanker[source]

Bases: sklearn.multiclass.OneVsRestClassifier

predict(self, X, top_k_list)[source]