Tiresias/config.yml at develop · RausellLab/Tiresias · GitHub

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
33
34
# Configure system resources
system:
  cpus: 2
  gpus: 0
  memory: 50 # in GB

# Set input datasets
data:
  edge_lists:
    - ./data/edge_lists/layer_0.tsv
    - ./data/edge_lists/layer_1.tsv
    - ./data/edge_lists/layer_2.tsv

  node_attributes: ./data/node_attributes.tsv
  train_node_labels: ./data/train_node_labels.tsv
  test_node_labels: ./data/test_node_labels.tsv

  # Set fraction of all data used for test file, if not test node labels is not indicated
  test_fraction: 0.1
# Enable or disable models
models:
  direct_neighbors: True
  label_spreading: True
  rwr: True
  bagging_gcn: True
  bagging_gcn_with_attributes: True
  rwr_m: True
  bagging_rgcn: True
  bagging_rgcn_with_attributes: True
  bagging_logistic_regression: True
  bagging_logistic_regression_with_attributes: True
  bagging_mlp: True
  bagging_mlp_with_attributes: True
  bagging_rgcn_with_embeddings: True