Module org.elasticsearch.xcore
Class InferenceHelpers
java.lang.Object
org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceHelpers
-
Nested Class Summary
Nested Classes -
Method Summary
Modifier and TypeMethodDescriptionstatic StringclassificationLabel(Integer inferenceValue, List<String> classificationLabels) decodeFeatureImportances(Map<String, String> processedFeatureToOriginalFeatureMap, Map<String, double[]> featureImportances) static voiddivMut(double[] xs, int v) static double[]sumDoubleArrays(double[] sumTo, double[] inc) static double[]sumDoubleArrays(double[] sumTo, double[] inc, int weight) static Doublestatic org.elasticsearch.core.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)
-
Method Details
-
topClasses
public static org.elasticsearch.core.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
-
toDouble
-
decodeFeatureImportances
-
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)
-