search

polsys / ennemi

Easy Nearest Neighbor Estimation of Mutual Information
https://polsys.github.io/ennemi/
MIT License
35 stars 12 forks source link

Support two-dimensional masks #37

Open polsys opened 4 years ago

polsys commented 4 years ago

Allow all masks that can be broadcasted to the shape of x. This would enable specifying a separate mask for each x variable, or even some more advanced scenarios.

For estimate_entropywith multidim=True, the masks should be combined, or two-dimensional masks be disallowed.

polsys commented 4 years ago

This is less necessary with the drop_nan support. I think this might still need more design and is not high-priority either. Moving to post 1.0.