com.microsoft.azure.synapse.ml.causal
OrthoForestDMLEstimator
Companion object OrthoForestDMLEstimator
class OrthoForestDMLEstimator extends Estimator[OrthoForestDMLModel] with ComplexParamsWritable with OrthoForestDMLParams with Wrappable with SynapseMLLogging with HasOutcomeCol
- Alphabetic
- By Inheritance
- OrthoForestDMLEstimator
- SynapseMLLogging
- Wrappable
- DotnetWrappable
- RWrappable
- PythonWrappable
- BaseWrappable
- OrthoForestDMLParams
- HasOutputCol
- HasMinSampleLeaf
- HasMaxDepth
- HasNumTrees
- DoubleMLParams
- HasParallelismInjected
- HasParallelism
- HasWeightCol
- HasMaxIter
- HasFeaturesCol
- HasOutcomeCol
- HasTreatmentCol
- ComplexParamsWritable
- MLWritable
- Estimator
- PipelineStage
- Logging
- Params
- Serializable
- Serializable
- Identifiable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Type Members
- type EstimatorWithPC = Estimator[_ <: Model[_] with HasPredictionCol] with HasPredictionCol
Value Members
-
final
def
clear(param: Param[_]): OrthoForestDMLEstimator.this.type
- Definition Classes
- Params
-
val
confidenceLevel: DoubleParam
- Definition Classes
- DoubleMLParams
-
val
confounderVecCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
def
copy(extra: ParamMap): Estimator[OrthoForestDMLModel]
- Definition Classes
- OrthoForestDMLEstimator → Estimator → PipelineStage → Params
-
def
dotnetAdditionalMethods: String
- Definition Classes
- DotnetWrappable
-
def
explainParam(param: Param[_]): String
- Definition Classes
- Params
-
def
explainParams(): String
- Definition Classes
- Params
-
final
def
extractParamMap(): ParamMap
- Definition Classes
- Params
-
final
def
extractParamMap(extra: ParamMap): ParamMap
- Definition Classes
- Params
-
val
featuresCol: Param[String]
The name of the features column
The name of the features column
- Definition Classes
- HasFeaturesCol
-
def
fit(dataset: Dataset[_]): OrthoForestDMLModel
Fits the OrthoForestDML model.
Fits the OrthoForestDML model.
- dataset
The input dataset to train.
- returns
The trained DoubleML model, from which you can get Ate and Ci values
- Definition Classes
- OrthoForestDMLEstimator → Estimator
-
def
fit(dataset: Dataset[_], paramMaps: Seq[ParamMap]): Seq[OrthoForestDMLModel]
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" )
-
def
fit(dataset: Dataset[_], paramMap: ParamMap): OrthoForestDMLModel
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" )
-
def
fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): OrthoForestDMLModel
- Definition Classes
- Estimator
- Annotations
- @Since( "2.0.0" ) @varargs()
-
final
def
get[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getConfidenceLevel: Double
- Definition Classes
- DoubleMLParams
-
def
getConfounderVecCol: String
- Definition Classes
- OrthoForestDMLParams
-
final
def
getDefault[T](param: Param[T]): Option[T]
- Definition Classes
- Params
-
def
getExecutionContextProxy: ExecutionContext
- Definition Classes
- HasParallelismInjected
-
def
getFeaturesCol: String
- Definition Classes
- HasFeaturesCol
-
def
getHeterogeneityVecCol: String
- Definition Classes
- OrthoForestDMLParams
-
def
getMaxDepth: Int
- Definition Classes
- HasMaxDepth
-
final
def
getMaxIter: Int
- Definition Classes
- HasMaxIter
-
def
getMinSamplesLeaf: Int
- Definition Classes
- HasMinSampleLeaf
-
def
getNumTrees: Int
- Definition Classes
- HasNumTrees
-
final
def
getOrDefault[T](param: Param[T]): T
- Definition Classes
- Params
-
def
getOutcomeCol: String
- Definition Classes
- HasOutcomeCol
-
def
getOutcomeModel: Estimator[_ <: Model[_]]
- Definition Classes
- DoubleMLParams
-
def
getOutcomeResidualCol: String
- Definition Classes
- OrthoForestDMLParams
-
def
getOutputCol: String
- Definition Classes
- HasOutputCol
-
def
getOutputHighCol: String
- Definition Classes
- OrthoForestDMLParams
-
def
getOutputLowCol: String
- Definition Classes
- OrthoForestDMLParams
-
def
getParallelism: Int
- Definition Classes
- HasParallelism
-
def
getParam(paramName: String): Param[Any]
- Definition Classes
- Params
-
def
getParamInfo(p: Param[_]): ParamInfo[_]
- Definition Classes
- BaseWrappable
-
def
getSampleSplitRatio: Array[Double]
- Definition Classes
- DoubleMLParams
-
def
getTreatmentCol: String
- Definition Classes
- HasTreatmentCol
-
def
getTreatmentModel: Estimator[_ <: Model[_]]
- Definition Classes
- DoubleMLParams
-
def
getTreatmentResidualCol: String
- Definition Classes
- OrthoForestDMLParams
-
def
getWeightCol: String
- Definition Classes
- HasWeightCol
-
final
def
hasDefault[T](param: Param[T]): Boolean
- Definition Classes
- Params
-
def
hasParam(paramName: String): Boolean
- Definition Classes
- Params
-
val
heterogeneityVecCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
final
def
isDefined(param: Param[_]): Boolean
- Definition Classes
- Params
-
final
def
isSet(param: Param[_]): Boolean
- Definition Classes
- Params
-
def
logClass(featureName: String): Unit
- Definition Classes
- SynapseMLLogging
-
def
logFit[T](f: ⇒ T, columns: Int): T
- Definition Classes
- SynapseMLLogging
-
def
logTransform[T](f: ⇒ T, columns: Int): T
- Definition Classes
- SynapseMLLogging
-
def
logVerb[T](verb: String, f: ⇒ T, columns: Option[Int] = None): T
- Definition Classes
- SynapseMLLogging
-
def
makeDotnetFile(conf: CodegenConfig): Unit
- Definition Classes
- DotnetWrappable
-
def
makePyFile(conf: CodegenConfig): Unit
- Definition Classes
- PythonWrappable
-
def
makeRFile(conf: CodegenConfig): Unit
- Definition Classes
- RWrappable
-
val
maxDepth: IntParam
- Definition Classes
- HasMaxDepth
-
final
val
maxIter: IntParam
- Definition Classes
- HasMaxIter
-
val
minSamplesLeaf: IntParam
- Definition Classes
- HasMinSampleLeaf
-
val
numTrees: IntParam
- Definition Classes
- HasNumTrees
-
val
outcomeCol: Param[String]
- Definition Classes
- HasOutcomeCol
-
val
outcomeModel: EstimatorParam
- Definition Classes
- DoubleMLParams
-
val
outcomeResidualCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
val
outputCol: Param[String]
The name of the output column
The name of the output column
- Definition Classes
- HasOutputCol
-
val
outputHighCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
val
outputLowCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
val
parallelism: IntParam
- Definition Classes
- HasParallelism
-
lazy val
params: Array[Param[_]]
- Definition Classes
- Params
-
def
pyAdditionalMethods: String
- Definition Classes
- PythonWrappable
-
def
pyInitFunc(): String
- Definition Classes
- PythonWrappable
-
val
sampleSplitRatio: DoubleArrayParam
- Definition Classes
- DoubleMLParams
-
def
save(path: String): Unit
- Definition Classes
- MLWritable
- Annotations
- @Since( "1.6.0" ) @throws( ... )
-
final
def
set[T](param: Param[T], value: T): OrthoForestDMLEstimator.this.type
- Definition Classes
- Params
-
def
setConfidenceLevel(value: Double): OrthoForestDMLEstimator.this.type
Set the higher bound percentile of ATE distribution.
Set the higher bound percentile of ATE distribution. Default is 0.975. lower bound value will be automatically calculated as 100*(1-confidenceLevel) That means by default we compute 95% confidence interval, it is [2.5%, 97.5%] percentile of ATE distribution
- Definition Classes
- DoubleMLParams
-
def
setConfounderVecCol(value: String): OrthoForestDMLEstimator.this.type
Set confounder vector column
Set confounder vector column
- Definition Classes
- OrthoForestDMLParams
-
def
setFeaturesCol(value: String): OrthoForestDMLEstimator.this.type
- Definition Classes
- HasFeaturesCol
-
def
setHeterogeneityVecCol(value: String): OrthoForestDMLEstimator.this.type
Set heterogeneity vector column
Set heterogeneity vector column
- Definition Classes
- OrthoForestDMLParams
-
def
setMaxDepth(value: Int): OrthoForestDMLEstimator.this.type
Set max depth of the trees to be used in the forest
Set max depth of the trees to be used in the forest
- Definition Classes
- HasMaxDepth
-
def
setMaxIter(value: Int): OrthoForestDMLEstimator.this.type
Set the maximum number of confidence interval bootstrapping iterations.
Set the maximum number of confidence interval bootstrapping iterations. Default is 1, which means it does not calculate confidence interval. To get Ci values please set a meaningful value
- Definition Classes
- DoubleMLParams
-
def
setMinSamplesLeaf(value: Int): OrthoForestDMLEstimator.this.type
Set number of samples in the leaf node of trees to be used in the forest
Set number of samples in the leaf node of trees to be used in the forest
- Definition Classes
- HasMinSampleLeaf
-
def
setNumTrees(value: Int): OrthoForestDMLEstimator.this.type
Set number of trees to be used in the forest
Set number of trees to be used in the forest
- Definition Classes
- HasNumTrees
-
def
setOutcomeCol(value: String): OrthoForestDMLEstimator.this.type
Set name of the column which will be used as outcome
Set name of the column which will be used as outcome
- Definition Classes
- HasOutcomeCol
-
def
setOutcomeModel(value: Estimator[_ <: Model[_]]): OrthoForestDMLEstimator.this.type
Set outcome model, it could be any model derived from 'org.apache.spark.ml.regression.Regressor' or 'org.apache.spark.ml.classification.ProbabilisticClassifier'
Set outcome model, it could be any model derived from 'org.apache.spark.ml.regression.Regressor' or 'org.apache.spark.ml.classification.ProbabilisticClassifier'
- Definition Classes
- DoubleMLParams
-
def
setOutcomeResidualCol(value: String): OrthoForestDMLEstimator.this.type
Set outcome residual column
Set outcome residual column
- Definition Classes
- OrthoForestDMLParams
-
def
setOutputCol(value: String): OrthoForestDMLEstimator.this.type
- Definition Classes
- HasOutputCol
-
def
setOutputHighCol(value: String): OrthoForestDMLEstimator.this.type
Set output column for effect upper bound
Set output column for effect upper bound
- Definition Classes
- OrthoForestDMLParams
-
def
setOutputLowCol(value: String): OrthoForestDMLEstimator.this.type
Set output column for effect lower bound
Set output column for effect lower bound
- Definition Classes
- OrthoForestDMLParams
-
def
setParallelism(value: Int): OrthoForestDMLEstimator.this.type
- Definition Classes
- DoubleMLParams
-
def
setSampleSplitRatio(value: Array[Double]): OrthoForestDMLEstimator.this.type
Set the sample split ratio, default is Array(0.5, 0.5)
Set the sample split ratio, default is Array(0.5, 0.5)
- Definition Classes
- DoubleMLParams
-
def
setTreatmentCol(value: String): OrthoForestDMLEstimator.this.type
Set name of the column which will be used as treatment
Set name of the column which will be used as treatment
- Definition Classes
- HasTreatmentCol
-
def
setTreatmentModel(value: Estimator[_ <: Model[_]]): OrthoForestDMLEstimator.this.type
Set treatment model, it could be any model derived from 'org.apache.spark.ml.regression.Regressor' or 'org.apache.spark.ml.classification.ProbabilisticClassifier'
Set treatment model, it could be any model derived from 'org.apache.spark.ml.regression.Regressor' or 'org.apache.spark.ml.classification.ProbabilisticClassifier'
- Definition Classes
- DoubleMLParams
-
def
setTreatmentResidualCol(value: String): OrthoForestDMLEstimator.this.type
Set treatment residual column
Set treatment residual column
- Definition Classes
- OrthoForestDMLParams
-
def
setWeightCol(value: String): OrthoForestDMLEstimator.this.type
- Definition Classes
- HasWeightCol
-
def
toString(): String
- Definition Classes
- Identifiable → AnyRef → Any
-
def
transformSchema(schema: StructType): StructType
- Definition Classes
- OrthoForestDMLEstimator → PipelineStage
-
val
treatmentCol: Param[String]
- Definition Classes
- HasTreatmentCol
-
val
treatmentModel: EstimatorParam
- Definition Classes
- DoubleMLParams
-
val
treatmentResidualCol: Param[String]
- Definition Classes
- OrthoForestDMLParams
-
val
uid: String
- Definition Classes
- OrthoForestDMLEstimator → SynapseMLLogging → Identifiable
-
val
weightCol: Param[String]
The name of the weight column
The name of the weight column
- Definition Classes
- HasWeightCol
-
def
write: MLWriter
- Definition Classes
- ComplexParamsWritable → MLWritable