adjust imports

autogl/datasets/__init__.py

@@ -136,10 +136,6 @@ __all__ = [
     "register_dataset",
     "build_dataset",
     "build_dataset_from_name",
-    "GatneDataset",
-    "GTNDataset",
-    "HANDataset",
-    "MatlabMatrix",
     "get_label_number",
     "random_splits_mask",
     "random_splits_mask_class",
@@ -147,4 +143,61 @@ __all__ = [
     "graph_set_fold_id",
     "graph_random_splits",
     "graph_get_split",
+    "AmazonComputersDataset",
+    "AmazonPhotoDataset",
+    "CoauthorPhysicsDataset",
+    "CoauthorCSDataset",
+    "CoraDataset",
+    "CiteSeerDataset",
+    "PubMedDataset",
+    "RedditDataset",
+    "MUTAGDataset",
+    "IMDBBinaryDataset",
+    "IMDBMultiDataset",
+    "CollabDataset",
+    "ProteinsDataset",
+    "REDDITBinary",
+    "REDDITMulti5K",
+    "REDDITMulti12K",
+    "PTCMRDataset",
+    "NCI1Dataset",
+    "ENZYMES",
+    "QM9Dataset",
+    "OGBNproductsDataset",
+    "OGBNproteinsDataset",
+    "OGBNarxivDataset",
+    "OGBNpapers100MDataset",
+    "OGBNmagDataset",
+    "OGBGmolhivDataset",
+    "OGBGmolpcbaDataset",
+    "OGBGppaDataset",
+    "OGBGcodeDataset",
+    "OGBLppaDataset",
+    "OGBLcollabDataset",
+    "OGBLddiDataset",
+    "OGBLcitationDataset",
+    "OGBLwikikgDataset",
+    "OGBLbiokgDataset",
+    "GatneDataset", 
+    "AmazonDataset", 
+    "TwitterDataset", 
+    "YouTubeDataset",
+    "GTNDataset", 
+    "ACM_GTNDataset", 
+    "DBLP_GTNDataset", 
+    "IMDB_GTNDataset",
+    "HANDataset", 
+    "ACM_HANDataset", 
+    "DBLP_HANDataset", 
+    "IMDB_HANDataset",
+    "MatlabMatrix",
+    "BlogcatalogDataset",
+    "WikipediaDataset",
+    "PPIDataset",
+    "ModelNet10",
+    "ModelNet40",
+    "ModelNet10Train",
+    "ModelNet10Test",
+    "ModelNet40Train",
+    "ModelNet40Test",
 ]

autogl/module/__init__.py

@@ -1,3 +1,12 @@
+from . import (
+    feature,
+    model,
+    train,
+    hpo,
+    nas,
+    ensemble
+)
+
 from .ensemble import *
 from .feature import *
 from .hpo import *

autogl/module/feature/__init__.py

@@ -1,5 +1,3 @@
-import importlib
-import os
 from .base import BaseFeatureAtom
 from .base import BaseFeatureEngineer
 
@@ -28,13 +26,83 @@ def register_feature(name):
 from .auto_feature import AutoFeatureEngineer
 from .base import BaseFeatureEngineer
 
-from .generators import BaseGenerator
-from .selectors import BaseSelector
+from .generators import (
+    BaseGenerator,
+    GeGraphlet,
+    GeEigen,
+    GePageRank,
+    register_pyg,
+    pygfunc,
+    PYGGenerator,
+    PYGLocalDegreeProfile,
+    PYGNormalizeFeatures,
+    PYGOneHotDegree
+)
 
-from .subgraph import BaseSubgraph
+from .selectors import (
+    BaseSelector,
+    SeFilterConstant, 
+    SeGBDT
+)
+
+from .subgraph import (
+    BaseSubgraph,
+    SgNetLSD,
+    register_nx,
+    NxSubgraph,
+    nxfunc,
+    NxLargeCliqueSize,
+    NxAverageClusteringApproximate,
+    NxDegreeAssortativityCoefficient,
+    NxDegreePearsonCorrelationCoefficient,
+    NxHasBridge,
+    NxGraphCliqueNumber,
+    NxGraphNumberOfCliques,
+    NxTransitivity,
+    NxAverageClustering,
+    NxIsConnected,
+    NxNumberConnectedComponents,
+    NxIsDistanceRegular,
+    NxLocalEfficiency,
+    NxGlobalEfficiency,
+    NxIsEulerian,
+)
 
 __all__ = [
     "BaseFeatureEngineer",
     "AutoFeatureEngineer",
     "BaseFeatureAtom",
+    "BaseGenerator",
+    "GeGraphlet",
+    "GeEigen",
+    "GePageRank",
+    "register_pyg",
+    "pygfunc",
+    "PYGGenerator",
+    "PYGLocalDegreeProfile",
+    "PYGNormalizeFeatures",
+    "PYGOneHotDegree",
+    "BaseSelector",
+    "SeFilterConstant",
+    "SeGBDT",
+    "BaseSubgraph",
+    "SgNetLSD",
+    "register_nx",
+    "NxSubgraph",
+    "nxfunc",
+    "NxLargeCliqueSize",
+    "NxAverageClusteringApproximate",
+    "NxDegreeAssortativityCoefficient",
+    "NxDegreePearsonCorrelationCoefficient",
+    "NxHasBridge",
+    "NxGraphCliqueNumber",
+    "NxGraphNumberOfCliques",
+    "NxTransitivity",
+    "NxAverageClustering",
+    "NxIsConnected",
+    "NxNumberConnectedComponents",
+    "NxIsDistanceRegular",
+    "NxLocalEfficiency",
+    "NxGlobalEfficiency",
+    "NxIsEulerian",
 ]

autogl/module/feature/generators/__init__.py

@@ -2,6 +2,17 @@ from .base import BaseGenerator
 from .graphlet import GeGraphlet
 from .eigen import GeEigen
 from .page_rank import GePageRank
-from .pyg import *
+from .pyg import register_pyg, PYGGenerator, pygfunc, PYGLocalDegreeProfile, PYGNormalizeFeatures, PYGOneHotDegree
 
-__all__ = ["BaseGenerator", "GeGraphlet", "GeEigen", "GePageRank"]
+__all__ = [
+    "BaseGenerator",
+    "GeGraphlet",
+    "GeEigen",
+    "GePageRank",
+    "register_pyg",
+    "pygfunc",
+    "PYGGenerator",
+    "PYGLocalDegreeProfile",
+    "PYGNormalizeFeatures",
+    "PYGOneHotDegree"
+]

autogl/module/feature/selectors/__init__.py

@@ -1,5 +1,3 @@
-import importlib
-import os
 from .base import BaseSelector
 from .se_filter_constant import SeFilterConstant
 from .se_gbdt import SeGBDT

autogl/module/feature/subgraph/__init__.py

@@ -1,4 +1,45 @@
 from .netlsd import SgNetLSD
 from .base import BaseSubgraph
+from .nx import (
+    register_nx,
+    NxSubgraph,
+    nxfunc,
+    NxLargeCliqueSize,
+    NxAverageClusteringApproximate,
+    NxDegreeAssortativityCoefficient,
+    NxDegreePearsonCorrelationCoefficient,
+    NxHasBridge,
+    NxGraphCliqueNumber,
+    NxGraphNumberOfCliques,
+    NxTransitivity,
+    NxAverageClustering,
+    NxIsConnected,
+    NxNumberConnectedComponents,
+    NxIsDistanceRegular,
+    NxLocalEfficiency,
+    NxGlobalEfficiency,
+    NxIsEulerian,
+)
 
-__all__ = ["SgNetLSD", "BaseSubgraph"]
+__all__ = [
+    "SgNetLSD",
+    "BaseSubgraph",
+    "register_nx",
+    "NxSubgraph",
+    "nxfunc",
+    "NxLargeCliqueSize",
+    "NxAverageClusteringApproximate",
+    "NxDegreeAssortativityCoefficient",
+    "NxDegreePearsonCorrelationCoefficient",
+    "NxHasBridge",
+    "NxGraphCliqueNumber",
+    "NxGraphNumberOfCliques",
+    "NxTransitivity",
+    "NxAverageClustering",
+    "NxIsConnected",
+    "NxNumberConnectedComponents",
+    "NxIsDistanceRegular",
+    "NxLocalEfficiency",
+    "NxGlobalEfficiency",
+    "NxIsEulerian",
+]

autogl/module/feature/subgraph/netlsd.py

@@ -1,6 +1,5 @@
 import netlsd
 from .base import BaseSubgraph
-import numpy as np
 import torch
 from .. import register_feature
 

autogl/module/feature/subgraph/nx.py

@@ -4,7 +4,6 @@ from networkx.algorithms.efficiency_measures import local_efficiency
 from networkx.algorithms.distance_regular import is_distance_regular
 from networkx.algorithms.components import number_connected_components
 from networkx.algorithms.components import is_connected
-from networkx.algorithms.cluster import average_clustering
 from networkx.algorithms.cluster import transitivity
 from networkx.algorithms.clique import graph_number_of_cliques
 from networkx.algorithms.clique import graph_clique_number
@@ -13,11 +12,8 @@ from networkx.algorithms.assortativity import degree_pearson_correlation_coeffic
 from networkx.algorithms.assortativity import degree_assortativity_coefficient
 from networkx.algorithms.approximation.clustering_coefficient import average_clustering
 from networkx.algorithms.approximation.clique import large_clique_size
-import netlsd
 from .base import BaseSubgraph
-import numpy as np
 import torch
-from functools import wraps
 from .. import register_feature
 
 NX_EXTRACTORS = []

autogl/module/feature/subgraph/stats.py

@@ -1 +0,0 @@
-import numpy as np

autogl/module/model/__init__.py

@@ -8,3 +8,15 @@ from .graph_saint import GraphSAINTAggregationModel
 from .gcn import AutoGCN
 from .gat import AutoGAT
 from .gin import AutoGIN
+
+__all__ = [
+    "ModelUniversalRegistry",
+    "register_model",
+    "BaseModel",
+    "AutoTopkpool",
+    "AutoSAGE",
+    "GraphSAINTAggregationModel",
+    "AutoGCN",
+    "AutoGAT",
+    "AutoGIN",
+]

autogl/module/nas/__init__.py

@@ -1,3 +1,9 @@
-from .algorithm import *
-from .estimator import *
-from .space import *
+from . import (
+    algorithm,
+    estimator,
+    space
+)
+
+from .algorithm import NAS_ALGO_DICT
+from .estimator import NAS_ESTIMATOR_DICT
+from .space import NAS_SPACE_DICT