exp_step1_batch.py

import numpy as np
from ddn3_extra import scan_r
from joblib import Parallel, delayed


def batch_run(l1_lst, l2_lst, n1, n2, n_node=100, ratio_diff=0.25, graph_type="random", n_group=0, n_rep=20):
    # repeat, method number, l2 number, l1 number, comm/diff, metric number
    # res = np.zeros((n_rep, 2, len(l2_lst), len(l1_lst), 2, 5))
    f_out = f"./res_ddn_jgl_{graph_type}_n_{n_rep}_n-node_{n_node}_{n1}p{n2}_group_{n_group}_ratio_{ratio_diff}.npz"
    print("Running ......", f_out, "\n")

    res = Parallel(n_jobs=8)(
        delayed(scan_r.scan_wrap_ddn_jgl)(
            l1_lst,
            l2_lst,
            n1,
            n2,
            n_node=n_node,
            ratio_diff=ratio_diff,
            graph_type=graph_type,
            n_group=n_group,
        )
        for _ in range(n_rep)
    )

    # save results
    res = np.array(res)
    np.savez(
        f_out,
        res=res,
        n_node=n_node,
        n1=n1,
        n2=n2,
        ratio_diff=ratio_diff,
        graph_type=graph_type,
        l1_lst=l1_lst,
        l2_lst=l2_lst,
    )


if __name__ == "__main__":
    l1_lst = np.arange(0.02, 1.0, 0.02)
    l2_lst = np.arange(0, 0.7, 0.1)
    # l2_lst = np.arange(0, 0.16, 0.025)
    # l2_lst = np.arange(0, 0.3, 0.05)

    # graph_type = "random"
    # graph_type = "scale-free-multi"
    # graph_type = "scale-free"
    # graph_type = "hub"
    # graph_type = "cluster"

    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=200, ratio_diff=0.25, graph_type=graph_type, n_group=8, n_rep=20)
    # batch_run(l1_lst, l2_lst, n1=50,  n2=50,  n_node=200, ratio_diff=0.25, graph_type=graph_type, n_group=8, n_rep=40)
    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=400, ratio_diff=0.25, graph_type=graph_type, n_group=8, n_rep=20)
    # batch_run(l1_lst, l2_lst, n1=50,  n2=50,  n_node=400, ratio_diff=0.25, graph_type=graph_type, n_group=8, n_rep=40)
    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=400, ratio_diff=0.25, graph_type=graph_type, n_group=16, n_rep=20)
    # batch_run(l1_lst, l2_lst, n1=50,  n2=50,  n_node=400, ratio_diff=0.25, graph_type=graph_type, n_group=16, n_rep=40)

    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=100, ratio_diff=0.25, graph_type=graph_type, n_group=4, n_rep=20)
    # batch_run(l1_lst, l2_lst, n1=50,  n2=50,  n_node=100, ratio_diff=0.25, graph_type=graph_type, n_group=4, n_rep=40)

    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=100, ratio_diff=0.25, graph_type=graph_type, n_group=3, n_rep=20)
    # batch_run(l1_lst, l2_lst, n1=50, n2=50, n_node=100, ratio_diff=0.25, graph_type=graph_type, n_group=3, n_rep=40)

    graph_type = "hub"
    batch_run(l1_lst, l2_lst, n1=100, n2=100, n_node=200, ratio_diff=0, graph_type=graph_type, n_group=5, n_rep=16)
    batch_run(l1_lst, l2_lst, n1=100, n2=100, n_node=200, ratio_diff=0.1, graph_type=graph_type, n_group=5, n_rep=16)
    batch_run(l1_lst, l2_lst, n1=100, n2=100, n_node=200, ratio_diff=0.25, graph_type=graph_type, n_group=5, n_rep=16)

    # graph_type = "random"
    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=200, ratio_diff=0, graph_type=graph_type, n_group=8, n_rep=16)
    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=200, ratio_diff=0.1, graph_type=graph_type, n_group=8, n_rep=16)
    # batch_run(l1_lst, l2_lst, n1=200, n2=200, n_node=200, ratio_diff=0.25, graph_type=graph_type, n_group=8, n_rep=16)