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)