Skip to content

Commit

Permalink
set DoubleMLEstimator transform as experimental
Browse files Browse the repository at this point in the history
  • Loading branch information
dylanw-oss committed Dec 16, 2022
1 parent 60f09e2 commit 44a5a35
Showing 1 changed file with 4 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ import com.microsoft.azure.synapse.ml.core.schema.{DatasetExtensions, SchemaCons
import com.microsoft.azure.synapse.ml.core.utils.StopWatch
import com.microsoft.azure.synapse.ml.logging.SynapseMLLogging
import com.microsoft.azure.synapse.ml.stages.DropColumns
import jdk.jfr.Experimental
import org.apache.commons.math3.stat.descriptive.rank.Percentile
import org.apache.spark.annotation.DeveloperApi
import org.apache.spark.ml.{ComplexParamsReadable, ComplexParamsWritable, Estimator, Model, Pipeline}
Expand Down Expand Up @@ -293,8 +294,10 @@ class DoubleMLModel(val uid: String)
override def copy(extra: ParamMap): DoubleMLModel = defaultCopy(extra)

/**
* DoubleMLEstimator transform does nothing by design and isn't supposed to be called by end user.
* :: Experimental ::
* DoubleMLEstimator transform function is still experimental, and its behavior could change in the future.
*/
@Experimental
override def transform(dataset: Dataset[_]): DataFrame = {
logTransform[DataFrame]({
dataset.toDF()
Expand Down

0 comments on commit 44a5a35

Please sign in to comment.