Esempio n. 1
0
def config():
    overwrite = None
    db_collection = None
    if db_collection is not None:
        ex.observers.append(
            seml.create_mongodb_observer(db_collection, overwrite=overwrite))

    # default params
    dataset = 'data/cora_ml.npz'
    n_per_class = 20
    seed = 42

    patience = 50
    max_epochs = 3000
    lr = 1e-3
    weight_decay = 1e-3

    model = 'GCN'
    n_hidden = 64
    p_dropout = 0.5

    pf_plus_adj = 0.0
    pf_minus_adj = 0.0

    pf_plus_att = 0.01
    pf_minus_att = 0.6

    n_samples_train = 1
    batch_size_train = 1

    n_samples_pre_eval = 10
    n_samples_eval = 1000
    batch_size_eval = 10

    mean_softmax = False
    conf_alpha = 0.01
    early_stopping = True

    save_dir = 'temp_dir'
Esempio n. 2
0
def config():
    overwrite = None
    db_collection = None
    if db_collection is not None:
        ex.observers.append(
            seml.create_mongodb_observer(db_collection, overwrite=overwrite))