public class RFormulaModel extends Model<RFormulaModel> implements MLWritable
RFormula
. Fitting is required to determine the factor levels of
formula terms.
param: resolvedFormula the fitted R formula. param: pipelineModel the fitted feature model, including factor to index mappings.
Modifier and Type | Method and Description |
---|---|
static Params |
clear(Param<?> param) |
RFormulaModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
static String |
explainParam(Param<?> param) |
static String |
explainParams() |
static ParamMap |
extractParamMap() |
static ParamMap |
extractParamMap(ParamMap extra) |
static Param<String> |
featuresCol() |
static BooleanParam |
forceIndexLabel() |
BooleanParam |
forceIndexLabel()
Force to index label whether it is numeric or string type.
|
static Param<String> |
formula() |
Param<String> |
formula()
R formula parameter.
|
static <T> scala.Option<T> |
get(Param<T> param) |
static <T> scala.Option<T> |
getDefault(Param<T> param) |
static String |
getFeaturesCol() |
static boolean |
getForceIndexLabel() |
boolean |
getForceIndexLabel() |
static String |
getFormula() |
String |
getFormula() |
static String |
getHandleInvalid() |
static String |
getLabelCol() |
static <T> T |
getOrDefault(Param<T> param) |
static Param<Object> |
getParam(String paramName) |
static String |
getStringIndexerOrderType() |
String |
getStringIndexerOrderType() |
static Param<String> |
handleInvalid() |
Param<String> |
handleInvalid()
Param for how to handle invalid data (unseen or NULL values) in features and label column
of string type.
|
static <T> boolean |
hasDefault(Param<T> param) |
boolean |
hasLabelCol(StructType schema) |
static boolean |
hasParam(String paramName) |
static boolean |
hasParent() |
static boolean |
isDefined(Param<?> param) |
static boolean |
isSet(Param<?> param) |
static Param<String> |
labelCol() |
static RFormulaModel |
load(String path) |
static Param<?>[] |
params() |
static void |
parent_$eq(Estimator<M> x$1) |
static Estimator<M> |
parent() |
static MLReader<RFormulaModel> |
read() |
static void |
save(String path) |
static <T> Params |
set(Param<T> param,
T value) |
static M |
setParent(Estimator<M> parent) |
static Param<String> |
stringIndexerOrderType() |
Param<String> |
stringIndexerOrderType()
Param for how to order categories of a string FEATURE column used by
StringIndexer . |
String |
toString() |
Dataset<Row> |
transform(Dataset<?> dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
transform, transform, transform
featuresCol, getFeaturesCol
getLabelCol, labelCol
getHandleInvalid
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
save
initializeLogging, initializeLogIfNecessary, initializeLogIfNecessary, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public static MLReader<RFormulaModel> read()
public static RFormulaModel load(String path)
public static Param<?>[] params()
public static String explainParam(Param<?> param)
public static String explainParams()
public static final boolean isSet(Param<?> param)
public static final boolean isDefined(Param<?> param)
public static boolean hasParam(String paramName)
public static Param<Object> getParam(String paramName)
public static final <T> scala.Option<T> get(Param<T> param)
public static final <T> T getOrDefault(Param<T> param)
public static final <T> scala.Option<T> getDefault(Param<T> param)
public static final <T> boolean hasDefault(Param<T> param)
public static final ParamMap extractParamMap()
public static Estimator<M> parent()
public static void parent_$eq(Estimator<M> x$1)
public static M setParent(Estimator<M> parent)
public static boolean hasParent()
public static final Param<String> featuresCol()
public static final String getFeaturesCol()
public static final Param<String> labelCol()
public static final String getLabelCol()
public static final String getHandleInvalid()
public static Param<String> formula()
public static String getFormula()
public static BooleanParam forceIndexLabel()
public static boolean getForceIndexLabel()
public static Param<String> handleInvalid()
public static final Param<String> stringIndexerOrderType()
public static String getStringIndexerOrderType()
public static void save(String path) throws java.io.IOException
java.io.IOException
public String uid()
Identifiable
uid
in interface Identifiable
public Dataset<Row> transform(Dataset<?> dataset)
Transformer
transform
in class Transformer
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Check transform validity and derive the output schema from the input schema.
We check validity for interactions between parameters during transformSchema
and
raise an exception if any parameter value is invalid. Parameter value checks which
do not depend on other parameters are handled by Param.validate()
.
Typical implementation should first conduct verification on schema change and parameter validity, including complex parameter interaction checks.
transformSchema
in class PipelineStage
schema
- (undocumented)public RFormulaModel copy(ParamMap extra)
Params
defaultCopy()
.copy
in interface Params
copy
in class Model<RFormulaModel>
extra
- (undocumented)public String toString()
toString
in interface Identifiable
toString
in class Object
public MLWriter write()
MLWritable
MLWriter
instance for this ML instance.write
in interface MLWritable
public Param<String> formula()
public String getFormula()
public BooleanParam forceIndexLabel()
public boolean getForceIndexLabel()
public Param<String> handleInvalid()
handleInvalid
in interface HasHandleInvalid
public Param<String> stringIndexerOrderType()
StringIndexer
.
The last category after ordering is dropped when encoding strings.
Supported options: 'frequencyDesc', 'frequencyAsc', 'alphabetDesc', 'alphabetAsc'.
The default value is 'frequencyDesc'. When the ordering is set to 'alphabetDesc', RFormula
drops the same category as R when encoding strings.
The options are explained using an example 'b', 'a', 'b', 'a', 'c', 'b'
:
+-----------------+---------------------------------------+----------------------------------+
| Option | Category mapped to 0 by StringIndexer | Category dropped by RFormula |
+-----------------+---------------------------------------+----------------------------------+
| 'frequencyDesc' | most frequent category ('b') | least frequent category ('c') |
| 'frequencyAsc' | least frequent category ('c') | most frequent category ('b') |
| 'alphabetDesc' | last alphabetical category ('c') | first alphabetical category ('a')|
| 'alphabetAsc' | first alphabetical category ('a') | last alphabetical category ('c') |
+-----------------+---------------------------------------+----------------------------------+
Note that this ordering option is NOT used for the label column. When the label column is
indexed, it uses the default descending frequency ordering in StringIndexer
.
public String getStringIndexerOrderType()
public boolean hasLabelCol(StructType schema)