pymc-devs/pymc

Add rewrite for Mixture when `comp_dists` can be "fused"

Open

#6 803 ouverte le 29 juin 2023

Voir sur GitHub
 (0 commentaires) (1 réaction) (0 assignés)Python (1 902 forks)batch import
feature requesthelp wantedlogprob

Métriques du dépôt

Stars
 (7 926 stars)
Métriques de merge PR
 (Merge moyen 11j 6h) (12 PRs mergées en 30 j)

Description

Description

The following distributions are equivalent:

import pymc as pm

pm.Mixture.dist(w=[0.5, 0.5], comp_dists=[pm.Normal.dist(-1), pm.Normal.dist(1)])
pm.Mixture.dist(w=[0.5, 0.5], comp_dists=pm.Normal.dist([-1, 1]))

But the second one is more efficient, because the logp is vectorized among a single batched Normal.

We could add a rewrite in the logprob_rewrites to convert the former to the second, so that users are not penalized from using the first form (which may be more intuitive for some).

Actually that sort of rewrite stack([rv1, rv2]) -> rv3 could be useful in many places in the logprob submodule

Guide contributeur

Direction de recherche
Étudiez le module logprob rewrites dans la base de code de PyMC (probablement dans pymc/logprob/rewrites.py ou similaire). L'objectif est d'ajouter une réécriture qui détecte lorsque les comp dists d'un Mixture sont toutes des instances de la même distribution (par exemple, Normal) avec des paramètres différents, et les remplace par une seule distribution en lot. Examinez les réécritures existantes pour identifier des modèles. Implémentez la transformation et ajoutez les tests correspondants pour garantir l'exactitude. Vérifiez si des PR ou commentaires liés fournissent des conseils ; actuellement aucun.
Stack technique
python
Domaine
backend
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Récente
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
Familiarity with PyMC's Mixture distributionUnderstanding of logp rewrites in PyMC
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30