summaryrefslogtreecommitdiff
path: root/scripts/config.py
blob: b23b090abf121b6fd25aa6e141bf9ce1676c3aad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""Central configuration for EC-SBM community detection analysis."""

import os

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
DATA_DIR = os.path.join(BASE_DIR, "data")
RESULTS_DIR = os.path.join(BASE_DIR, "results")

NETWORKS = {
    "polblogs": {
        "edge_tsv": os.path.join(DATA_DIR, "polblogs", "edge.tsv"),
        "com_gt_tsv": os.path.join(DATA_DIR, "polblogs", "com_gt.tsv"),
    },
    "topology": {
        "edge_tsv": os.path.join(DATA_DIR, "topology", "edge.tsv"),
        "com_gt_tsv": os.path.join(DATA_DIR, "topology", "com_gt.tsv"),
    },
    "internet_as": {
        "edge_tsv": os.path.join(DATA_DIR, "internet_as", "edge.tsv"),
        "com_gt_tsv": os.path.join(DATA_DIR, "internet_as", "com_gt.tsv"),
    },
}

METHODS = [
    {"name": "leiden_mod", "type": "leiden", "quality": "modularity"},
    {"name": "leiden_cpm_01", "type": "leiden", "quality": "cpm", "resolution": 0.1},
    {"name": "leiden_cpm_001", "type": "leiden", "quality": "cpm", "resolution": 0.01},
    {"name": "infomap", "type": "infomap"},
    {"name": "graphtool_sbm", "type": "graphtool_sbm"},
]

SEED = 42