objective_iterative_rewiring

Table of contents

  1. Overview
  2. Basic Usage

Overview

This Optuna objective tunes the parameters of the iterative rewiring pipeline and SelectiveGCN model. It samples hyperparameters such as the number of rewiring iterations, SGC settings and learning rates, then calls run_iterative_bridge_experiment.

Basic Usage

study = optuna.create_study(direction="minimize")
study.optimize(lambda t: objective_iterative_rewiring(
    t,
    g,
    best_gcn_params,
    all_matrices,
    device="cuda"
), n_trials=50)