From 05f09c5f7b461d1822f4e8043d371aaa232e21e0 Mon Sep 17 00:00:00 2001 From: Frozenmad Date: Mon, 21 Dec 2020 21:07:44 +0800 Subject: [PATCH] change default config to gcn small --- examples/node_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/node_classification.py b/examples/node_classification.py index 1e7a61e..af7334d 100644 --- a/examples/node_classification.py +++ b/examples/node_classification.py @@ -16,7 +16,7 @@ if __name__ == '__main__': from argparse import ArgumentParser parser = ArgumentParser() parser.add_argument('--dataset', default='cora', type=str) - parser.add_argument('--configs', type=str, default='../configs/node_classification.yaml') + parser.add_argument('--configs', type=str, default='../configs/nodeclf_gat_benchmark_small.yml') # following arguments will override parameters in the config file parser.add_argument('--hpo', type=str, default='random') parser.add_argument('--max_eval', type=int, default=5)