java.lang.Object
org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceHelpers

public final class InferenceHelpers extends Object
  • 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

      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)