@@ -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
3030parameters :
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
3939split_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
4848dataloader_params :
49- batch_size : 16
49+ batch_size : 1
5050 num_workers : 0
5151 pin_memory : False
0 commit comments