Skip to content

Commit

Permalink
Merge pull request #1 from aidos-lab/mantra-dev-typing
Browse files Browse the repository at this point in the history 
Mantra dev typing
  • Loading branch information
@ErnstRoell
ErnstRoell authored Jun 5, 2024
2 parents b952d5d + 367f4dd commit bee1fb3
Show file tree
Hide file tree
Showing 725 changed files with 64,727 additions and 149 deletions.
3 changes: 2 additions & 1 deletion .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,5 @@ test*.*
**/__pycache__
/lightning_logs/
k-simplex*
RandomWalks*
RandomWalks*
results
16 changes: 16 additions & 0 deletions configs/gat_betti_numbers_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 9
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gat
use_stratified: false
16 changes: 16 additions & 0 deletions configs/gat_betti_numbers_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 8
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gat
use_stratified: false
16 changes: 16 additions & 0 deletions configs/gat_name_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 9
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gat
use_stratified: true
16 changes: 16 additions & 0 deletions configs/gat_name_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 8
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gat
use_stratified: true
16 changes: 16 additions & 0 deletions configs/gat_orientability_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 9
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gat
use_stratified: true
16 changes: 16 additions & 0 deletions configs/gat_orientability_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
conf_model:
hidden_channels: 64
num_heads: 4
num_hidden_layers: 3
num_node_features: 8
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gat
use_stratified: true
15 changes: 15 additions & 0 deletions configs/gcn_betti_numbers_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gcn
use_stratified: false
15 changes: 15 additions & 0 deletions configs/gcn_betti_numbers_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gcn
use_stratified: false
15 changes: 15 additions & 0 deletions configs/gcn_name_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gcn
use_stratified: true
15 changes: 15 additions & 0 deletions configs/gcn_name_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gcn
use_stratified: true
15 changes: 15 additions & 0 deletions configs/gcn_orientability_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: gcn
use_stratified: true
15 changes: 15 additions & 0 deletions configs/gcn_orientability_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: gcn
use_stratified: true
15 changes: 15 additions & 0 deletions configs/tag_betti_numbers_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: tag
use_stratified: false
15 changes: 15 additions & 0 deletions configs/tag_betti_numbers_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: tag
use_stratified: false
15 changes: 15 additions & 0 deletions configs/tag_name_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: tag
use_stratified: true
15 changes: 15 additions & 0 deletions configs/tag_name_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: tag
use_stratified: true
15 changes: 15 additions & 0 deletions configs/tag_orientability_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: tag
use_stratified: true
15 changes: 15 additions & 0 deletions configs/tag_orientability_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: tag
use_stratified: true
15 changes: 15 additions & 0 deletions configs/transfconv_betti_numbers_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: transfconv
use_stratified: false
15 changes: 15 additions & 0 deletions configs/transfconv_betti_numbers_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 3
learning_rate: 0.01
seed: 1234
task_type: betti_numbers
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: transfconv
use_stratified: false
15 changes: 15 additions & 0 deletions configs/transfconv_name_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: transfconv
use_stratified: true
15 changes: 15 additions & 0 deletions configs/transfconv_name_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 5
learning_rate: 0.01
seed: 1234
task_type: name
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: transfconv
use_stratified: true
15 changes: 15 additions & 0 deletions configs/transfconv_orientability_degree_transform.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 9
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: degree_transform
type_model: transfconv
use_stratified: true
15 changes: 15 additions & 0 deletions configs/transfconv_orientability_random_node_features.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
conf_model:
hidden_channels: 64
num_hidden_layers: 3
num_node_features: 8
out_channels: 1
learning_rate: 0.01
seed: 1234
task_type: orientability
trainer_config:
accelerator: auto
log_every_n_steps: 1
max_epochs: 50
transforms: random_node_features
type_model: transfconv
use_stratified: true
Loading

0 comments on commit bee1fb3

Please sign in to comment.