Lightgbm

Initialization of the LightGBM integration.

LightGBMIntegration

Bases: Integration

Definition of lightgbm integration for ZenML.

Source code in src/zenml/integrations/lightgbm/__init__.py

class LightGBMIntegration(Integration):
    """Definition of lightgbm integration for ZenML."""

    NAME = LIGHTGBM
    REQUIREMENTS = ["lightgbm>=1.0.0"]
    APT_PACKAGES = ["libgomp1"]

    @classmethod
    def activate(cls) -> None:
        """Activates the integration."""
        from zenml.integrations.lightgbm import materializers  # noqa

activate() classmethod

Activates the integration.

Source code in src/zenml/integrations/lightgbm/__init__.py

@classmethod
def activate(cls) -> None:
    """Activates the integration."""
    from zenml.integrations.lightgbm import materializers  # noqa