public interface PredictorParams extends Params, HasLabelCol, HasFeaturesCol, HasPredictionCol
Trait for parameters for prediction (regression and classification).
NOTE: This is currently private[spark] but will be made public later once it is stabilized.
Modifier and Type | Method and Description |
---|---|
StructType |
validateAndTransformSchema(StructType schema,
ParamMap paramMap,
boolean fitting,
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
StructType validateAndTransformSchema(StructType schema, ParamMap paramMap, boolean fitting, DataType featuresDataType)
schema
- input schemaparamMap
- additional parametersfitting
- whether this is in fittingfeaturesDataType
- SQL DataType for FeaturesType.
E.g., VectorUDT
for vector features.