Browse Source

change default config to gcn small

tags/v0.3.1
Frozenmad 5 years ago
parent
commit
05f09c5f7b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      examples/node_classification.py

+ 1
- 1
examples/node_classification.py View File

@@ -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)


Loading…
Cancel
Save