"""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
)