Performance improvements to weighted_mi

bowman-lab / enspara

Modeling molecular ensembles with scalable data structures and parallel computing

https://enspara.readthedocs.io

GNU General Public License v3.0

33 stars 16 forks source link

Closed justinrporter closed 5 years ago

justinrporter commented 5 years ago

weighted_mi can now compute MIs for features that take on more than two states.
Use matrix multiplication rather than iteration to compute joint probabilities in weighted_mi
Use meshgrid to compute channel capacities.

~300x speed improvement for weighted_mi.