Interface FeatureImportanceClassifier

    • Method Detail

      • getFeatureImportances

        double[] getFeatureImportances​(boolean normalize)
        Obtain the current importance for each feature.
        Returns:
        array containing the importance/score estimated for each feature
      • getTopKFeatures

        int[] getTopKFeatures​(int k,
                              boolean normalize)
        The output is a double array where values indicates the original feature index and the order of the array its ranking. The size of this array is expected to be less than the complete set of features.
        Parameters:
        k -
        normalize -
        Returns:
        the k features with the highest scores.