Evalml: Ajouter un tracé de dépendance partielle

Créé le 28 juil. 2020  ·  3Commentaires  ·  Source: alteryx/evalml

Tous les 3 commentaires

Une chose qui ne m'est pas venue à l'esprit auparavant, c'est que nous pouvons réellement utiliser des valeurs de shap pour faire le tracé des dépendances. https://slundberg.github.io/shap/notebooks/plots/dependence_plot.html

Je soupçonne que nous commençons toujours par ajouter des tracés de deps partiels traditionnels, mais je voulais partager cela car c'est intéressant et quelque chose à garder à l'esprit lorsque nous développons cette API.

@kmax12 Merci pour ce partage ! Je suis d'accord qu'il serait logique de commencer par ajouter des tracés de dépendance partiels traditionnels, j'ai donc déposé # 1068 pour suivre l'utilisation des valeurs SHAP pour créer des tracés de dépendance afin de m'assurer que cela ne se perde pas lorsque ce problème est fermé.

Discuté avec @angela97lin . Plats à emporter :

  • Pour l'instant, nous prendrons en charge les pipelines. Nous n'ajouterons pas de couverture de test / de documentation pour accepter les estimateurs en entrée, à moins que cela ne s'avère "juste travailler" et ne prenne que quelques heures supplémentaires pour ajouter une couverture de test unitaire
  • Nous ne générerons que des tracés 1D, pas des tracés d'interaction 2D
  • API :
def partial_dependence(pipeline, X, feature_name, grid_resolution=None):
    ...
def graph_partial_dependence(pipeline, X, feature_name, grid_resolution=None):
    ...
Cette page vous a été utile?
0 / 5 - 0 notes