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

Function test_agc_cora

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

Source from the content-addressed store, hash-verified

69
70
71def test_agc_cora():
72 args = get_default_args_agc(dataset="cora", model="agc", mw="agc_mw", dw="node_classification_dw")
73 args.model_type = "both"
74 args.cluster_method = "spectral"
75 args.max_iter = 1
76 ret = train(args)
77 assert ret["nmi"] >= 0
78
79
80def test_daegc_cora():

Calls 2

trainFunction · 0.90
get_default_args_agcFunction · 0.85

Tested by

no test coverage detected