Source code for synapse.ml.lightgbm.LightGBMRegressionModel

# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.

from synapse.ml.lightgbm._LightGBMRegressionModel import _LightGBMRegressionModel
from synapse.ml.lightgbm.mixin import LightGBMModelMixin
from pyspark import SparkContext
from pyspark.ml.common import inherit_doc
from pyspark.ml.wrapper import JavaParams
from synapse.ml.core.serialize.java_params_patch import *

[docs]@inherit_doc class LightGBMRegressionModel(LightGBMModelMixin, _LightGBMRegressionModel):
[docs] @staticmethod def loadNativeModelFromFile(filename): """ Load the model from a native LightGBM text file. """ ctx = SparkContext._active_spark_context loader = ctx._jvm.com.microsoft.azure.synapse.ml.lightgbm.LightGBMRegressionModel java_model = loader.loadNativeModelFromFile(filename) return JavaParams._from_java(java_model)
[docs] @staticmethod def loadNativeModelFromString(model): """ Load the model from a native LightGBM model string. """ ctx = SparkContext._active_spark_context loader = ctx._jvm.com.microsoft.azure.synapse.ml.lightgbm.LightGBMRegressionModel java_model = loader.loadNativeModelFromString(model) return JavaParams._from_java(java_model)