()
| 106 | |
| 107 | |
| 108 | def test_diffpool_imdb_binary(): |
| 109 | args = get_default_args_graph_clf(dataset="imdb-b", model="diffpool") |
| 110 | args = add_diffpool_args(args) |
| 111 | args.batch_size = 100 |
| 112 | args.train_ratio = 0.6 |
| 113 | args.test_ratio = 0.2 |
| 114 | ret = train(args) |
| 115 | assert ret["test_acc"] > 0 |
| 116 | |
| 117 | |
| 118 | def test_sortpool_mutag(): |
no test coverage detected