| 12345678910111213141516171819 |
- import numpy as np
- from scipy.stats import dirichlet
- import ternary
- scale = 1
- x = np.load("votes_distrib.npz")
- alphas = x["alphas"].mean(axis=0)
- def prob(p):
- print(p)
- return dirichlet.pdf(p, alphas)
- figure, tax = ternary.figure(scale=scale)
- tax.heatmapf(prob, boundary=True, style="triangular")
- tax.boundary(linewidth=2.0)
- tax.set_title("Shannon Entropy Heatmap")
- tax.savefig("votes_distrib.eps")
|
