Module org.elasticsearch.xcore

Package org.elasticsearch.xpack.core.ml.dataframe.analyses

Class Regression

java.lang.Object

org.elasticsearch.xpack.core.ml.dataframe.analyses.Regression

All Implemented Interfaces:

NamedWriteable, Writeable, ToXContent, ToXContentObject, DataFrameAnalysis

public class Regression
extends Object
implements DataFrameAnalysis

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	Regression.LossFunction

Nested classes/interfaces inherited from interface org.elasticsearch.xpack.core.ml.dataframe.analyses.DataFrameAnalysis

DataFrameAnalysis.FieldInfo

Nested classes/interfaces inherited from interface org.elasticsearch.xcontent.ToXContent

ToXContent.DelegatingMapParams, ToXContent.MapParams, ToXContent.Params

Nested classes/interfaces inherited from interface org.elasticsearch.common.io.stream.Writeable

Writeable.Reader<V>, Writeable.Writer<V>

Field Summary

Fields

Modifier and Type	Field	Description
static final ParseField	DEPENDENT_VARIABLE
static final ParseField	EARLY_STOPPING_ENABLED
static final ParseField	FEATURE_PROCESSORS
static final ParseField	LOSS_FUNCTION
static final ParseField	LOSS_FUNCTION_PARAMETER
static final ParseField	NAME
static final ParseField	PREDICTION_FIELD_NAME
static final ParseField	RANDOMIZE_SEED
static final ParseField	TRAINING_PERCENT

Fields inherited from interface org.elasticsearch.xcontent.ToXContent

EMPTY, EMPTY_PARAMS

Constructor Summary

Constructors

Constructor	Description
Regression(String dependentVariable)
Regression(String dependentVariable, BoostedTreeParams boostedTreeParams, String predictionFieldName, Double trainingPercent, Long randomizeSeed, Regression.LossFunction lossFunction, Double lossFunctionParameter, List<PreProcessor> featureProcessors, Boolean earlyStoppingEnabled)
Regression(StreamInput in)

Method Summary

Modifier and Type	Method	Description
boolean	equals(Object o)
static String	extractJobIdFromStateDoc(String stateDocId)
static Regression	fromXContent(XContentParser parser, boolean ignoreUnknownFields)
Set<String>	getAllowedCategoricalTypes(String fieldName)
BoostedTreeParams	getBoostedTreeParams()
String	getDependentVariable()
Boolean	getEarlyStoppingEnabled()
List<PreProcessor>	getFeatureProcessors()
List<FieldCardinalityConstraint>	getFieldCardinalityConstraints()
Regression.LossFunction	getLossFunction()
Double	getLossFunctionParameter()
Map<String,Object>	getParams(DataFrameAnalysis.FieldInfo fieldInfo)
String	getPredictionFieldName()
List<String>	getProgressPhases()	Returns the progress phases the analysis goes through in order
long	getRandomizeSeed()
List<RequiredField>	getRequiredFields()
Map<String,Object>	getResultMappings(String resultsFieldName, FieldCapabilitiesResponse fieldCapabilitiesResponse)	Returns fields for which the mappings should be either predefined or copied from source index to destination index.
String	getStateDocIdPrefix(String jobId)	Returns the document id prefix for the analysis state
double	getTrainingPercent()
String	getWriteableName()
int	hashCode()
InferenceConfig	inferenceConfig(DataFrameAnalysis.FieldInfo fieldInfo)
boolean	persistsState()
boolean	supportsCategoricalFields()
boolean	supportsInference()
boolean	supportsMissingValues()
XContentBuilder	toXContent(XContentBuilder builder, ToXContent.Params params)
void	writeTo(StreamOutput out)

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.elasticsearch.xcontent.ToXContentObject

isFragment

Field Details

NAME

public static final ParseField NAME

DEPENDENT_VARIABLE

public static final ParseField DEPENDENT_VARIABLE

PREDICTION_FIELD_NAME

public static final ParseField PREDICTION_FIELD_NAME

TRAINING_PERCENT

public static final ParseField TRAINING_PERCENT

RANDOMIZE_SEED

public static final ParseField RANDOMIZE_SEED

LOSS_FUNCTION

public static final ParseField LOSS_FUNCTION

LOSS_FUNCTION_PARAMETER

public static final ParseField LOSS_FUNCTION_PARAMETER

FEATURE_PROCESSORS

public static final ParseField FEATURE_PROCESSORS

EARLY_STOPPING_ENABLED

public static final ParseField EARLY_STOPPING_ENABLED

Constructor Details

Regression

public Regression(String dependentVariable,
 BoostedTreeParams boostedTreeParams,
 @Nullable
 String predictionFieldName,
 @Nullable
 Double trainingPercent,
 @Nullable
 Long randomizeSeed,
 @Nullable
 Regression.LossFunction lossFunction,
 @Nullable
 Double lossFunctionParameter,
 @Nullable
 List<PreProcessor> featureProcessors,
 @Nullable
 Boolean earlyStoppingEnabled)

Regression

public Regression(String dependentVariable)

Regression

public Regression(StreamInput in)
           throws IOException

Throws:

IOException

Method Details

fromXContent

public static Regression fromXContent(XContentParser parser,
 boolean ignoreUnknownFields)

getDependentVariable

public String getDependentVariable()

getBoostedTreeParams

public BoostedTreeParams getBoostedTreeParams()

getPredictionFieldName

public String getPredictionFieldName()

getTrainingPercent

public double getTrainingPercent()

Specified by:

getTrainingPercent in interface DataFrameAnalysis

Returns:

the percentage of data to use for training

getRandomizeSeed

public long getRandomizeSeed()

getLossFunction

public Regression.LossFunction getLossFunction()

getLossFunctionParameter

public Double getLossFunctionParameter()

getFeatureProcessors

public List<PreProcessor> getFeatureProcessors()

getEarlyStoppingEnabled

public Boolean getEarlyStoppingEnabled()

getWriteableName

public String getWriteableName()

Specified by:

getWriteableName in interface NamedWriteable

writeTo

public void writeTo(StreamOutput out)
             throws IOException

Specified by:

writeTo in interface Writeable

Throws:

IOException

toXContent

public XContentBuilder toXContent(XContentBuilder builder,
 ToXContent.Params params)
                           throws IOException

Specified by:

toXContent in interface ToXContent

Throws:

IOException

getParams

public Map<String,Object> getParams(DataFrameAnalysis.FieldInfo fieldInfo)

Specified by:

getParams in interface DataFrameAnalysis

Parameters:

fieldInfo - Information about the fields like types and cardinalities

Returns:

The analysis parameters as a map

supportsCategoricalFields

public boolean supportsCategoricalFields()

Specified by:

supportsCategoricalFields in interface DataFrameAnalysis

Returns:

true if this analysis supports fields with categorical values (i.e. text, keyword, ip)

getAllowedCategoricalTypes

public Set<String> getAllowedCategoricalTypes(String fieldName)

Specified by:

getAllowedCategoricalTypes in interface DataFrameAnalysis

Parameters:

fieldName - field for which the allowed categorical types should be returned

Returns:

The types treated as categorical for the given field

getRequiredFields

public List<RequiredField> getRequiredFields()

Specified by:

getRequiredFields in interface DataFrameAnalysis

Returns:

The names and types of the fields that analyzed documents must have for the analysis to operate

getFieldCardinalityConstraints

public List<FieldCardinalityConstraint> getFieldCardinalityConstraints()

Specified by:

getFieldCardinalityConstraints in interface DataFrameAnalysis

Returns:

List containing cardinality constraints for the selected (analysis-specific) fields

getResultMappings

public Map<String,Object> getResultMappings(String resultsFieldName,
 FieldCapabilitiesResponse fieldCapabilitiesResponse)

Description copied from interface: DataFrameAnalysis

Returns fields for which the mappings should be either predefined or copied from source index to destination index.

Specified by:

getResultMappings in interface DataFrameAnalysis

Parameters:

resultsFieldName - name of the results field under which all the results are stored

fieldCapabilitiesResponse - field capabilities fetched for this analysis' required fields

Returns:

Map containing fields for which the mappings should be handled explicitly

supportsMissingValues

public boolean supportsMissingValues()

Specified by:

supportsMissingValues in interface DataFrameAnalysis

Returns:

true if this analysis supports data frame rows with missing values

persistsState

public boolean persistsState()

Specified by:

persistsState in interface DataFrameAnalysis

Returns:

true if this analysis persists state that can later be used to restore from a given point

getStateDocIdPrefix

public String getStateDocIdPrefix(String jobId)

Description copied from interface: DataFrameAnalysis

Returns the document id prefix for the analysis state

Specified by:

getStateDocIdPrefix in interface DataFrameAnalysis

getProgressPhases

public List<String> getProgressPhases()

Description copied from interface: DataFrameAnalysis

Returns the progress phases the analysis goes through in order

Specified by:

getProgressPhases in interface DataFrameAnalysis

inferenceConfig

public InferenceConfig inferenceConfig(DataFrameAnalysis.FieldInfo fieldInfo)

Specified by:

inferenceConfig in interface DataFrameAnalysis

Returns:

the analysis inference config or null if inference is not supported

supportsInference

public boolean supportsInference()

Specified by:

supportsInference in interface DataFrameAnalysis

Returns:

true if this analysis trains a model that can be used for inference

extractJobIdFromStateDoc

public static String extractJobIdFromStateDoc(String stateDocId)

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object