GM&GED problem

ml4co_kit/__init__.py

@@ -46,9 +46,13 @@
 from .task import TaskBase, TASK_TYPE
 
 # Graph Task
-from .task import GraphTaskBase
+from .task import GraphTaskBase 
 from .task import MClTask, MCutTask, MISTask, MVCTask
 
+# Graph Set Task
+from .task import Graph, Graph, GraphSetTaskBase
+from .task import GMTask, GEDTask
+
 # Routing Task
 from .task import RoutingTaskBase, DISTANCE_TYPE, ROUND_TYPE
 from .task import ATSPTask, CVRPTask, OPTask, PCTSPTask, SPCTSPTask, TSPTask
@@ -76,6 +80,14 @@
 )
 from .generator import MClGenerator, MCutGenerator, MISGenerator, MVCGenerator
 
+# GraphSet Generator
+from .generator import (
+    GraphFeatureGenerator, GraphSetGeneratorBase, 
+    GRAPH_TYPE, GRAPH_FEATURE_TYPE, 
+)
+from .generator import GMGenerator
+from .generator import GEDGenerator
+
 # Portfolio Generator
 from .generator import PortfolioGeneratorBase, PO_TYPE, PortfolioDistributionArgs
 from .generator import MinVarPOGenerator, MaxRetPOGenerator, MOPOGenerator
@@ -105,15 +117,15 @@
 # Solver (not use torch backend)
 from .solver import (
     ConcordeSolver, GAEAXSolver, GpDegreeSolver, GurobiSolver, 
-    HGSSolver, ILSSolver, InsertionSolver, ISCOSolver, KaMISSolver, 
-    LcDegreeSolver, LKHSolver, ORSolver, RandomSolver, SCIPSolver
+    HGSSolver, ILSSolver, InsertionSolver, KaMISSolver, 
+    LcDegreeSolver, LKHSolver, ORSolver, SMSolver, IPFPSolver, RRWMSolver
 )
 
 # Solver (use torch backend)
 if env_checker.check_gnn4co():
     from .solver import GNN4COSolver
 if env_checker.check_torch():
-    from .solver import NeuroLKHSolver, RLSASolver
+    from .solver import NeuroLKHSolver, RLSASolver, NGMSolver, AStarSolver, GennAStarSolver
 
 
 ####################################################
@@ -153,12 +165,18 @@
     MClWrapper, MCutWrapper, MISWrapper, MVCWrapper
 )
 
+# Graph Set Problems
+from .wrapper import(
+    GMWrapper, GEDWrapper
+)
+
 # Portfolio Problems
 from .wrapper import (
     MinVarPOWrapper, MaxRetPOWrapper, MOPOWrapper
 )
 
 
+
 ####################################################
 #                    Learning                      #
 ####################################################
@@ -169,9 +187,5 @@
     )
 
 
-####################################################
-#                Version and Author                #
-####################################################
-
-__version__ = "0.4.2"
+__version__ = "1.0.0"
 __author__ = "SJTU-ReThinkLab"

ml4co_kit/generator/__init__.py

@@ -25,6 +25,13 @@
 from .graph.mis import MISGenerator
 from .graph.mvc import MVCGenerator
 
+# Graph Set Generator
+from .graphset.base import (
+    GraphSetGeneratorBase, GRAPH_TYPE, GRAPH_FEATURE_TYPE, GraphFeatureGenerator
+)
+from .graphset.gm import GMGenerator
+from .graphset.ged import GEDGenerator
+
 # Portfolio Generator
 from .portfolio.base import (
     PortfolioGeneratorBase, PO_TYPE, PortfolioDistributionArgs