monarch-initiative/N2V

"""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",
    task_name="Edge Prediction",
    expected_parent_class=SklearnEdgePredictionAdapter
)