ispras
diff --git a/‎experiments/various_tasks.py‎
Lines changed: 5 additions & 2 deletions b/‎experiments/various_tasks.py‎
Lines changed: 5 additions & 2 deletions
diff --git a/‎gnn_aid/datasets/__pycache__/__init__.cpython-311.pyc‎
0 Bytes b/‎gnn_aid/datasets/__pycache__/__init__.cpython-311.pyc‎
0 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/dataset_converter.cpython-311.pyc‎
-67 Bytes b/‎gnn_aid/datasets/__pycache__/dataset_converter.cpython-311.pyc‎
-67 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/dataset_info.cpython-311.pyc‎
-48 Bytes b/‎gnn_aid/datasets/__pycache__/dataset_info.cpython-311.pyc‎
-48 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/dataset_stats.cpython-311.pyc‎
-41 Bytes b/‎gnn_aid/datasets/__pycache__/dataset_stats.cpython-311.pyc‎
-41 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/datasets_manager.cpython-311.pyc‎
-28 Bytes b/‎gnn_aid/datasets/__pycache__/datasets_manager.cpython-311.pyc‎
-28 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/gen_dataset.cpython-311.pyc‎
-139 Bytes b/‎gnn_aid/datasets/__pycache__/gen_dataset.cpython-311.pyc‎
-139 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/heterographs.cpython-311.pyc‎
-27.6 KB b/‎gnn_aid/datasets/__pycache__/heterographs.cpython-311.pyc‎
-27.6 KB
diff --git a/‎gnn_aid/datasets/__pycache__/known_format_datasets.cpython-311.pyc‎
-98 Bytes b/‎gnn_aid/datasets/__pycache__/known_format_datasets.cpython-311.pyc‎
-98 Bytes
diff --git a/‎gnn_aid/datasets/__pycache__/ptg_datasets.cpython-311.pyc‎
-46 Bytes b/‎gnn_aid/datasets/__pycache__/ptg_datasets.cpython-311.pyc‎
-46 Bytes
@@ -134,7 +134,6 @@ def link_prediction():
                 ]
             )))
 
-
     manager_config = ConfigPattern(
         _config_class="ModelManagerConfig",
         _config_kwargs={
@@ -154,7 +153,7 @@ def link_prediction():
         }
     )
 
-    steps_epochs = 1
+    steps_epochs = 3
     my_device = device('cuda' if torch.cuda.is_available() else 'cpu')
     gnn_model_manager = FrameworkGNNModelManager(
         gnn=gnn,
@@ -174,6 +173,10 @@ def link_prediction():
     )
     print("Training was successful")
 
+    res = gnn_model_manager.run_model(
+        gen_dataset=gen_dataset,
+        mask='all'
+    )
     res = gnn_model_manager.evaluate_model(
         gen_dataset=gen_dataset,
         metrics=[Metric("Accuracy", mask="test")]
Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,6 @@ def link_prediction():`
`134`	`134`	`]`
`135`	`135`	`)))`
`136`	`136`
`137`		`-`
`138`	`137`	`manager_config = ConfigPattern(`
`139`	`138`	`_config_class="ModelManagerConfig",`
`140`	`139`	`_config_kwargs={`
`@@ -154,7 +153,7 @@ def link_prediction():`
`154`	`153`	`}`
`155`	`154`	`)`
`156`	`155`
`157`		`- steps_epochs = 1`
	`156`	`+ steps_epochs = 3`
`158`	`157`	`my_device = device('cuda' if torch.cuda.is_available() else 'cpu')`
`159`	`158`	`gnn_model_manager = FrameworkGNNModelManager(`
`160`	`159`	`gnn=gnn,`
`@@ -174,6 +173,10 @@ def link_prediction():`
`174`	`173`	`)`
`175`	`174`	`print("Training was successful")`
`176`	`175`
	`176`	`+ res = gnn_model_manager.run_model(`
	`177`	`+ gen_dataset=gen_dataset,`
	`178`	`+ mask='all'`
	`179`	`+ )`
`177`	`180`	`res = gnn_model_manager.evaluate_model(`
`178`	`181`	`gen_dataset=gen_dataset,`
`179`	`182`	`metrics=[Metric("Accuracy", mask="test")]`