Module org.elasticsearch.xcore

Package org.elasticsearch.xpack.core.ml.inference.trainedmodel

Class InferenceHelpers

java.lang.Object

org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceHelpers

public final class InferenceHelpers
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	InferenceHelpers.TopClassificationValue

Method Summary

Modifier and Type	Method	Description
static String	classificationLabel(Integer inferenceValue, List<String> classificationLabels)
static Map<String,double[]>	decodeFeatureImportances(Map<String,String> processedFeatureToOriginalFeatureMap, Map<String,double[]> featureImportances)
static void	divMut(double[] xs, int v)
static double[]	sumDoubleArrays(double[] sumTo, double[] inc)
static double[]	sumDoubleArrays(double[] sumTo, double[] inc, int weight)
static Double	toDouble(Object value)
static Tuple<InferenceHelpers.TopClassificationValue,List<TopClassEntry>>	topClasses(double[] probabilities, List<String> classificationLabels, double[] classificationWeights, int numToInclude, PredictionFieldType predictionFieldType)
static List<ClassificationFeatureImportance>	transformFeatureImportanceClassification(Map<String,double[]> featureImportance, List<String> classificationLabels, PredictionFieldType predictionFieldType)
static List<RegressionFeatureImportance>	transformFeatureImportanceRegression(Map<String,double[]> featureImportance)

Methods inherited from class java.lang.Object

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

Method Details

topClasses

public static Tuple<InferenceHelpers.TopClassificationValue,List<TopClassEntry>> topClasses(double[] probabilities,
 List<String> classificationLabels,
 @Nullable
 double[] classificationWeights,
 int numToInclude,
 PredictionFieldType predictionFieldType)

Returns:

Tuple of the highest scored index and the top classes

classificationLabel

public static String classificationLabel(Integer inferenceValue,
 @Nullable
 List<String> classificationLabels)

toDouble

public static Double toDouble(Object value)

decodeFeatureImportances

public static Map<String,double[]> decodeFeatureImportances(Map<String,String> processedFeatureToOriginalFeatureMap,
 Map<String,double[]> featureImportances)

transformFeatureImportanceRegression

public static List<RegressionFeatureImportance> transformFeatureImportanceRegression(Map<String,double[]> featureImportance)

transformFeatureImportanceClassification

public static List<ClassificationFeatureImportance> transformFeatureImportanceClassification(Map<String,double[]> featureImportance,
 @Nullable
 List<String> classificationLabels,
 @Nullable
 PredictionFieldType predictionFieldType)

sumDoubleArrays

public static double[] sumDoubleArrays(double[] sumTo,
 double[] inc)

sumDoubleArrays

public static double[] sumDoubleArrays(double[] sumTo,
 double[] inc,
 int weight)

divMut

public static void divMut(double[] xs,
 int v)