public interface ProbabilisticClassifierParams extends ClassifierParams, HasProbabilityCol
Modifier and Type | Method and Description |
---|---|
org.apache.spark.sql.types.StructType |
validateAndTransformSchema(org.apache.spark.sql.types.StructType schema,
ParamMap paramMap,
boolean fitting,
org.apache.spark.sql.types.DataType featuresDataType)
Validates and transforms the input schema with the provided param map.
|
getLabelCol, labelCol
featuresCol, getFeaturesCol
getPredictionCol, predictionCol
addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate
uid
getRawPredictionCol, rawPredictionCol
getProbabilityCol, probabilityCol
org.apache.spark.sql.types.StructType validateAndTransformSchema(org.apache.spark.sql.types.StructType schema, ParamMap paramMap, boolean fitting, org.apache.spark.sql.types.DataType featuresDataType)
PredictorParams
validateAndTransformSchema
in interface ClassifierParams
validateAndTransformSchema
in interface PredictorParams
schema
- input schemaparamMap
- additional parametersfitting
- whether this is in fittingfeaturesDataType
- SQL DataType for FeaturesType.
E.g., VectorUDT
for vector features.