You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

hetero_node_classification.py 368 B

12345678910111213
  1. import os
  2. os.environ["AUTOGL_BACKEND"] = 'dgl'
  3. from autogl.datasets import build_dataset_from_name
  4. from autogl.solver import AutoHeteroNodeClassifier
  5. if __name__ == '__main__':
  6. acm = build_dataset_from_name("hetero-acm-han")
  7. solver = AutoHeteroNodeClassifier(max_evals=10)
  8. solver.fit(acm)
  9. acc = solver.evaluate(metric='acc')
  10. print("acc: ", acc)