Extends
Methods
copy(extra) → {object}
Parameters:
Name | Type | Description |
---|---|---|
extra |
module:eclairjs/ml/param.ParamMap |
- Inherited From:
- Source:
Returns:
- Type
- object
featuresCol() → {module:eclairjs/ml/param.Param}
Param for features column name.
- Source:
Returns:
getFeaturesCol() → {string}
- Source:
Returns:
- Type
- string
getLabelCol() → {Promise.<string>}
- Source:
Returns:
- Type
- Promise.<string>
getPredictionCol() → {Promise.<string>}
- Source:
Returns:
- Type
- Promise.<string>
hasParent() → {Promise.<boolean>}
Indicates whether this Model has a corresponding parent.
- Inherited From:
- Source:
Returns:
- Type
- Promise.<boolean>
labelCol() → {module:eclairjs/ml/param.Param}
Param for label column name.
- Source:
Returns:
numFeatures() → {Promise.<number>}
Returns the number of features the model was trained on. If unknown, returns -1
- Source:
Returns:
- Type
- Promise.<number>
parent() → {module:eclairjs/ml.Estimator}
- Inherited From:
- Source:
Returns:
predictionCol() → {module:eclairjs/ml/param.Param}
Param for prediction column name.
- Source:
Returns:
setFeaturesCol(value) → {object}
Parameters:
Name | Type | Description |
---|---|---|
value |
string |
- Source:
Returns:
- Type
- object
setParent(parent) → {object}
Sets the parent of this model.
Parameters:
Name | Type | Description |
---|---|---|
parent |
module:eclairjs/ml.Estimator |
- Inherited From:
- Source:
Returns:
- Type
- object
setPredictionCol(value) → {object}
Parameters:
Name | Type | Description |
---|---|---|
value |
string |
- Source:
Returns:
- Type
- object
transform(dataset) → {module:eclairjs/sql.Dataset}
Transforms dataset by reading from featuresCol, calling [[predict()]], and storing
the predictions as a new column predictionCol.
Parameters:
Name | Type | Description |
---|---|---|
dataset |
module:eclairjs/sql.Dataset | input dataset |
- Overrides:
- Source:
Returns:
transformed dataset with [[predictionCol]] of type [[Double]]
transformSchema(schema) → {module:eclairjs/sql/types.StructType}
Parameters:
Name | Type | Description |
---|---|---|
schema |
module:eclairjs/sql/types.StructType |
- Source:
Returns:
validateAndTransformSchema(schema, fitting, featuresDataType) → {module:eclairjs/sql/types.StructType}
Validates and transforms the input schema with the provided param map.
Parameters:
Name | Type | Description |
---|---|---|
schema |
module:eclairjs/sql/types.StructType | |
fitting |
boolean | whether this is in fitting |
featuresDataType |
module:eclairjs/sql/types.DataType | SQL DataType for FeaturesType. E.g., module:eclairjs/sql/types.VectorUDTfor vector features |
- Source: