MCPcopy
hub / github.com/THUDM/CogDL / test_daegc_cora

Function test_daegc_cora

tests/tasks/test_attributed_graph_clustering.py:80–85  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

78
79
80def test_daegc_cora():
81 args = get_default_args_agc(dataset="cora", model="daegc", mw="daegc_mw", dw="node_classification_dw")
82 args.model_type = "both"
83 args.cluster_method = "kmeans"
84 ret = train(args)
85 assert ret["nmi"] >= 0
86
87
88def test_gae_cora():

Calls 2

trainFunction · 0.90
get_default_args_agcFunction · 0.85

Tested by

no test coverage detected