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

Function test_gin_imdb_binary

tests/tasks/test_graph_classification.py:93–98  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

91
92
93def test_gin_imdb_binary():
94 args = get_default_args_graph_clf(dataset="imdb-b", model="gin")
95 args = add_gin_args(args)
96 args.degree_node_features = True
97 ret = train(args)
98 assert ret["test_acc"] > 0
99
100
101def test_gin_proteins():

Callers 1

Calls 3

trainFunction · 0.90
add_gin_argsFunction · 0.85

Tested by

no test coverage detected