public class KMeansModel extends Model<KMeansModel>
param: parentModel a model trained by spark.mllib.clustering.KMeans.
Modifier and Type | Method and Description |
---|---|
Vector[] |
clusterCenters() |
KMeansModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
java.lang.String |
getInitMode() |
int |
getInitSteps() |
int |
getK() |
Param<java.lang.String> |
initMode()
Param for the initialization algorithm.
|
IntParam |
initSteps()
Param for the number of steps for the k-means|| initialization mode.
|
IntParam |
k()
Set the number of clusters to create (k).
|
DataFrame |
transform(DataFrame dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
java.lang.String |
uid()
An immutable unique ID for the object and its derivatives.
|
StructType |
validateAndTransformSchema(StructType schema)
Validates and transforms the input schema.
|
transform, transform, transform
transformSchema
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn, validateParams
toString
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public java.lang.String uid()
Identifiable
uid
in interface Identifiable
public KMeansModel copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Model<KMeansModel>
extra
- (undocumented)defaultCopy()
public DataFrame transform(DataFrame dataset)
Transformer
transform
in class Transformer
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Derives the output schema from the input schema.
transformSchema
in class PipelineStage
schema
- (undocumented)public Vector[] clusterCenters()
public IntParam k()
public int getK()
public Param<java.lang.String> initMode()
public java.lang.String getInitMode()
public IntParam initSteps()
public int getInitSteps()
public StructType validateAndTransformSchema(StructType schema)
schema
- input schema