AutoGL/test/performance/node_classification/dgl/helper.py

def get_encoder_decoder_hp(model='gin', decoder=None):
    if model == 'gin':
        model_hp = {
            "num_layers": 5,
            "hidden": [64],
            "act": "relu",
            "eps": "False",
            "mlp_layers": 2,
            "neighbor_pooling_type": "sum"
        }
    elif model == 'gat':
        model_hp = {
            # hp from model
            "num_layers": 2,
            "hidden": [8],
            "heads": 8,
            "dropout": 0.6,
            "act": "relu",
        }
    elif model == 'gcn':
        model_hp = {
            "num_layers": 2,
            "hidden": [16],
            "dropout": 0.5,
            "act": "relu"
        }
    elif model == 'sage':
        model_hp = {
            "num_layers": 2,
            "hidden": [64],
            "dropout": 0.5,
            "act": "relu",
            "agg": "gcn",
        }
    elif model == 'topk':
        model_hp = {
            "num_layers": 5,
            "hidden": [64, 64, 64, 64]
        }
    
    return model_hp, {}