Partager via


PermutationFeatureImportanceExtensions.PermutationFeatureImportanceNonCalibrated Méthode

Définition

Permutation Feature Importance (PFI) pour la classification binaire.

public static System.Collections.Immutable.ImmutableDictionary<string,Microsoft.ML.Data.BinaryClassificationMetricsStatistics> PermutationFeatureImportanceNonCalibrated (this Microsoft.ML.BinaryClassificationCatalog catalog, Microsoft.ML.ITransformer model, Microsoft.ML.IDataView data, string labelColumnName = "Label", bool useFeatureWeightFilter = false, int? numberOfExamplesToUse = default, int permutationCount = 1);
static member PermutationFeatureImportanceNonCalibrated : Microsoft.ML.BinaryClassificationCatalog * Microsoft.ML.ITransformer * Microsoft.ML.IDataView * string * bool * Nullable<int> * int -> System.Collections.Immutable.ImmutableDictionary<string, Microsoft.ML.Data.BinaryClassificationMetricsStatistics>
<Extension()>
Public Function PermutationFeatureImportanceNonCalibrated (catalog As BinaryClassificationCatalog, model As ITransformer, data As IDataView, Optional labelColumnName As String = "Label", Optional useFeatureWeightFilter As Boolean = false, Optional numberOfExamplesToUse As Nullable(Of Integer) = Nothing, Optional permutationCount As Integer = 1) As ImmutableDictionary(Of String, BinaryClassificationMetricsStatistics)

Paramètres

catalog
BinaryClassificationCatalog

Catalogue de classification binaire.

model
ITransformer

Modèle sur lequel évaluer l’importance des caractéristiques.

data
IDataView

Jeu de données d’évaluation.

labelColumnName
String

Nom de colonne d’étiquette. Les données de colonne doivent être Boolean.

useFeatureWeightFilter
Boolean

Utilisez le poids des caractéristiques pour préfiltrer les fonctionnalités.

numberOfExamplesToUse
Nullable<Int32>

Limitez le nombre d’exemples à évaluer. signifie jusqu’à ~2 exemples bln de sera utilisé.

permutationCount
Int32

Nombre de permutations à effectuer.

Retours

Dictionnaire mappant chaque fonctionnalité à ses « contributions » par fonctionnalité au score.

Exemples

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.ML;

namespace Samples.Dynamic.Trainers.BinaryClassification
{
    public static class PermutationFeatureImportance
    {
        public static void Example()
        {
            // Create a new context for ML.NET operations. It can be used for
            // exception tracking and logging, as a catalog of available operations
            // and as the source of randomness.
            var mlContext = new MLContext(seed: 1);

            // Create sample data.
            var samples = GenerateData();

            // Load the sample data as an IDataView.
            var data = mlContext.Data.LoadFromEnumerable(samples);

            // Define a training pipeline that concatenates features into a vector,
            // normalizes them, and then trains a linear model.
            var featureColumns =
                new string[] { nameof(Data.Feature1), nameof(Data.Feature2) };
            var pipeline = mlContext.Transforms
                .Concatenate("Features", featureColumns)
                .Append(mlContext.Transforms.NormalizeMinMax("Features"))
                .Append(mlContext.BinaryClassification.Trainers
                .SdcaLogisticRegression());

            // Fit the pipeline to the data.
            var model = pipeline.Fit(data);

            // Transform the dataset.
            var transformedData = model.Transform(data);

            // Extract the predictor.
            var linearPredictor = model.LastTransformer;

            // Compute the permutation metrics for the linear model using the
            // normalized data.
            var permutationMetrics = mlContext.BinaryClassification
                .PermutationFeatureImportance(linearPredictor, transformedData,
                permutationCount: 30);

            // Now let's look at which features are most important to the model
            // overall. Get the feature indices sorted by their impact on AUC.
            var sortedIndices = permutationMetrics
                .Select((metrics, index) => new { index, metrics.AreaUnderRocCurve })
                .OrderByDescending(
                feature => Math.Abs(feature.AreaUnderRocCurve.Mean))
                .Select(feature => feature.index);

            Console.WriteLine("Feature\tModel Weight\tChange in AUC"
                + "\t95% Confidence in the Mean Change in AUC");
            var auc = permutationMetrics.Select(x => x.AreaUnderRocCurve).ToArray();
            foreach (int i in sortedIndices)
            {
                Console.WriteLine("{0}\t{1:0.00}\t{2:G4}\t{3:G4}",
                    featureColumns[i],
                    linearPredictor.Model.SubModel.Weights[i],
                    auc[i].Mean,
                    1.96 * auc[i].StandardError);
            }

            // Expected output:
            //  Feature     Model Weight Change in AUC  95% Confidence in the Mean Change in AUC
            //  Feature2        35.15     -0.387        0.002015
            //  Feature1        17.94     -0.1514       0.0008963
        }

        private class Data
        {
            public bool Label { get; set; }

            public float Feature1 { get; set; }

            public float Feature2 { get; set; }
        }

        /// <summary>
        /// Generate an enumerable of Data objects, creating the label as a simple
        /// linear combination of the features.
        /// </summary>
        /// <param name="nExamples">The number of examples.</param>
        /// <param name="bias">The bias, or offset, in the calculation of the label.
        /// </param>
        /// <param name="weight1">The weight to multiply the first feature with to
        /// compute the label.</param>
        /// <param name="weight2">The weight to multiply the second feature with to
        /// compute the label.</param>
        /// <param name="seed">The seed for generating feature values and label
        /// noise.</param>
        /// <returns>An enumerable of Data objects.</returns>
        private static IEnumerable<Data> GenerateData(int nExamples = 10000,
            double bias = 0, double weight1 = 1, double weight2 = 2, int seed = 1)
        {
            var rng = new Random(seed);
            for (int i = 0; i < nExamples; i++)
            {
                var data = new Data
                {
                    Feature1 = (float)(rng.Next(10) * (rng.NextDouble() - 0.5)),
                    Feature2 = (float)(rng.Next(10) * (rng.NextDouble() - 0.5)),
                };

                // Create a noisy label.
                var value = (float)(bias + weight1 * data.Feature1 + weight2 *
                    data.Feature2 + rng.NextDouble() - 0.5);

                data.Label = Sigmoid(value) > 0.5;
                yield return data;
            }
        }

        private static double Sigmoid(double x) => 1.0 / (1.0 + Math.Exp(-1 * x));
    }
}

Remarques

L’importance des caractéristiques de permutation (PFI) est une technique permettant de déterminer l’importance globale des fonctionnalités dans un modèle Machine Learning entraîné. PFI est une technique simple mais puissante motivée par Breiman dans son papier de forêt aléatoire, section 10 (Breiman. « Forêts aléatoires ». Machine Learning, 2001.) L’avantage de la méthode PFI est qu’il est indépendant du modèle - il fonctionne avec n’importe quel modèle qui peut être évalué - et il peut utiliser n’importe quel jeu de données, pas seulement le jeu d’entraînement, pour calculer les métriques d’importance des caractéristiques.

PFI fonctionne en prenant un jeu de données étiqueté, en choisissant une fonctionnalité et en permutant les valeurs de cette fonctionnalité dans tous les exemples, afin que chaque exemple ait désormais une valeur aléatoire pour la fonctionnalité et les valeurs d’origine pour toutes les autres fonctionnalités. La métrique d’évaluation (par exemple, AUC) est ensuite calculée pour ce jeu de données modifié, et la modification de la métrique d’évaluation du jeu de données d’origine est calculée. Plus la modification de la métrique d’évaluation est importante, plus la fonctionnalité est importante pour le modèle. PFI fonctionne en effectuant cette analyse de permutation sur toutes les fonctionnalités d’un modèle, l’une après l’autre.

Dans cette implémentation, PFI calcule la modification de toutes les métriques d’évaluation de classification binaire possibles pour chaque fonctionnalité, et un ImmutableArray objet BinaryClassificationMetrics est retourné. Consultez l’exemple ci-dessous pour obtenir un exemple d’utilisation de ces résultats pour analyser l’importance des caractéristiques d’un modèle.

S’applique à