Lucas Gautheron 3 mēneši atpakaļ
vecāks
revīzija
2ea7b376be
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 2 2
      plot_votes_distrib.py

+ 2 - 2
plot_votes_distrib.py

@@ -8,10 +8,10 @@ x = np.load("votes_distrib.npz")
 alphas = x["alphas"].mean(axis=0)
 
 def prob(p):
-    return dirichlet.logpdf(quantiles, alphas)
+    return dirichlet.pdf(p, alphas)
 
 figure, tax = ternary.figure(scale=scale)
-tax.heatmapf(shannon_entropy, boundary=True, style="triangular")
+tax.heatmapf(prob, boundary=True, style="triangular")
 tax.boundary(linewidth=2.0)
 tax.set_title("Shannon Entropy Heatmap")