org.apache.spark.ml.regression
Class RegressionModel<FeaturesType,M extends RegressionModel<FeaturesType,M>>

Object
  org.apache.spark.ml.PipelineStage
      org.apache.spark.ml.Transformer
          org.apache.spark.ml.Model<M>
              org.apache.spark.ml.PredictionModel<FeaturesType,M>
                  org.apache.spark.ml.regression.RegressionModel<FeaturesType,M>

All Implemented Interfaces:

java.io.Serializable, Logging, Params

Direct Known Subclasses:

LinearRegressionModel

public abstract class RegressionModel<FeaturesType,M extends RegressionModel<FeaturesType,M>>
extends PredictionModel<FeaturesType,M>

:: DeveloperApi ::

Model produced by a Regressor.

See Also:

Serialized Form

Constructor Summary

RegressionModel()

Method Summary

StructType

validateAndTransformSchema(StructType schema, boolean fitting, DataType featuresDataType) Validates and transforms the input schema with the provided param map.

Methods inherited from class org.apache.spark.ml.PredictionModel

setFeaturesCol, setPredictionCol, transform, transformSchema

Methods inherited from class org.apache.spark.ml.Model

copy, hasParent, parent, setParent

Methods inherited from class org.apache.spark.ml.Transformer

transform, transform, transform

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

RegressionModel

public RegressionModel()

Method Detail

validateAndTransformSchema

public StructType validateAndTransformSchema(StructType schema,
                                             boolean fitting,
                                             DataType featuresDataType)

Validates and transforms the input schema with the provided param map.

Parameters:

schema - input schema

fitting - whether this is in fitting

featuresDataType - SQL DataType for FeaturesType. E.g., VectorUDT for vector features.

Returns:

output schema