"""Submodule providing node embedding models implemented in Karate Club.""" from embiggen.utils.abstract_models import build_init, AbstractEmbeddingModel build_init( module_library_names="karateclub", formatted_library_name="Karate Club", expected_parent_class=AbstractEmbeddingModel )
"""Submodule providing edge-label prediction models based on Sklearn Models.""" from embiggen.edge_label_prediction.edge_label_prediction_model import AbstractEdgeLabelPredictionModel from embiggen.utils.abstract_models import build_init build_init(module_library_names="tensorflow", formatted_library_name="TensorFlow", expected_parent_class=AbstractEdgeLabelPredictionModel)
"""Submodule providing edge prediction models based on Sklearn Models.""" from embiggen.edge_prediction.edge_prediction_sklearn.sklearn_edge_prediction_adapter import SklearnEdgePredictionAdapter from embiggen.utils.abstract_models import build_init build_init(module_library_names="sklearn", formatted_library_name="scikit-learn", expected_parent_class=SklearnEdgePredictionAdapter)
"""Submodule providing node embedding models implemented in Ensmallen in Rust.""" from embiggen.utils.abstract_models import build_init, AbstractEmbeddingModel build_init( module_library_names="ensmallen", formatted_library_name="Ensmallen", expected_parent_class=AbstractEmbeddingModel )
"""Module with graph embedding models based on PyKeen.""" from embiggen.utils.abstract_models import build_init, AbstractEmbeddingModel build_init(module_library_names=["torch", "pykeen"], formatted_library_name="PyKeen", expected_parent_class=AbstractEmbeddingModel)
"""Submodule providing node embedding models implemented in Ensmallen in Rust.""" from embiggen.edge_prediction.edge_prediction_model import AbstractEdgePredictionModel from embiggen.utils.abstract_models import build_init build_init( module_library_names="ensmallen", formatted_library_name="Ensmallen", expected_parent_class=AbstractEdgePredictionModel )