Skip to content

Releases: mlelarge/graph_neural_net

QAP Regular ErdosRenyi 50 0.2 0.1 (2023)

25 Aug 09:18
Compare
Choose a tag to compare

Updated code to make it compatible with PyTorch Lightning

QAP Regular ErdosRenyi 50 0.2 0.15

21 Jan 16:52
e35691c
Compare
Choose a tag to compare

Config:
"data": {"num_examples_train": 20000, "num_examples_val": 1000, "generative_model": "Regular", "noise_model": "ErdosRenyi", "edge_density": 0.2, "n_vertices": 50, "vertex_proba": 1.0, "noise": 0.15, "path_dataset": "dataset"},
"train": {"epoch": 50, "batch_size": 32, "lr": 0.0001, "scheduler_step": 5, "scheduler_decay": 0.9, "print_freq": 100, "loss_reduction": "mean"}, "arch": {"arch": "Siamese_Model", "model_name": "Simple_Node_Embedding", "num_blocks": 2, "original_features_num": 2, "in_features": 64, "out_features": 64, "depth_of_mlp": 3}