pymc-devs/pymc

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

Open

#6.803 aberto em 29 de jun. de 2023

Ver no GitHub
 (0 comments) (1 reaction) (0 assignees)Python (1.902 forks)batch import
feature requesthelp wantedlogprob

Métricas do repositório

Stars
 (7.926 stars)
Métricas de merge de PR
 (Mesclagem média 11d 6h) (12 fundiu PRs em 30d)

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

Guia do colaborador

Direção de pesquisa
Investigue o módulo logprob rewrites na base de código do PyMC (provavelmente em pymc/logprob/rewrites.py ou similar). O objetivo é adicionar uma reescrita que detecte quando os comp dists de uma Mixture são todos instâncias da mesma distribuição (por exemplo, Normal) com parâmetros diferentes, e os substitua por uma única distribuição em lote. Examine as reescritas existentes para identificar padrões. Implemente a transformação e adicione testes correspondentes para garantir a correção. Verifique se há PRs ou comentários vinculados que forneçam orientação; atualmente nenhum.
Pilha de tecnologia
python
Domain
backend
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Recente
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
Familiarity with PyMC's Mixture distributionUnderstanding of logp rewrites in PyMC
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
30