Skip to content

Commit b809c7c

Browse files
Add transductive experiment yaml instead of triangle counting task
1 parent f9fd491 commit b809c7c

1 file changed

Lines changed: 18 additions & 18 deletions

File tree

configs/dataset/graph/GraphUniverse_TC.yaml renamed to configs/dataset/graph/graphuniverse_transductive.yaml

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -7,37 +7,37 @@ loader:
77
data_name: GraphUniverse
88
data_dir: ${paths.data_dir}/${dataset.loader.parameters.data_domain}/${dataset.loader.parameters.data_type}
99
generation_parameters:
10-
task: triangle_counting
10+
task: community_detection
1111
universe_parameters:
12-
K: 20
12+
K: 10
1313
feature_dim: 15
1414
center_variance: 0.2
15-
cluster_variance: 0.4
16-
edge_propensity_variance: 1.0
15+
cluster_variance: 0.5
16+
edge_propensity_variance: 0.5
1717
seed: 42
1818
family_parameters:
19-
n_graphs: 1000
20-
n_nodes_range: [50, 200]
21-
n_communities_range: [3, 7]
22-
homophily_range: [0.4, 0.8]
23-
avg_degree_range: [1.0, 2.0]
24-
degree_separation_range: [0.5, 1.0]
25-
power_law_exponent_range: [1.5, 2.5]
19+
n_graphs: 1
20+
n_nodes_range: [5000, 5000]
21+
n_communities_range: [10, 10]
22+
homophily_range: [0.5, 0.5]
23+
avg_degree_range: [2.5, 2.5]
24+
degree_separation_range: [0.5, 0.5]
25+
power_law_exponent_range: [2.5, 2.5]
2626
seed: ${dataset.loader.parameters.generation_parameters.universe_parameters.seed}
2727

2828

2929
# Dataset parameters
3030
parameters:
3131
num_features: ${dataset.loader.parameters.generation_parameters.universe_parameters.feature_dim}
32-
num_classes: 1
33-
task: regression
34-
loss_type: mae
35-
monitor_metric: mae
36-
task_level: graph
32+
num_classes: ${dataset.loader.parameters.generation_parameters.universe_parameters.K}
33+
task: classification
34+
loss_type: cross_entropy
35+
monitor_metric: accuracy
36+
task_level: node
3737

3838
#splits
3939
split_params:
40-
learning_setting: inductive
40+
learning_setting: transductive
4141
data_split_dir: ${dataset.loader.parameters.data_dir}/data_splits
4242
data_seed: 0
4343
split_type: random #'k-fold' # either "k-fold" or "random" strategies
@@ -46,6 +46,6 @@ split_params:
4646

4747
# Dataloader parameters
4848
dataloader_params:
49-
batch_size: 16
49+
batch_size: 1
5050
num_workers: 0
5151
pin_memory: False

0 commit comments

Comments
 (0)