Zeel commited on
Commit
7e3eb7e
·
1 Parent(s): 50e41fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -15,7 +15,7 @@ beta = st.slider('Beta', 0.5, 5.0, 0.5)
15
  theta = np.linspace(0.01,0.99,100)
16
  #rc('font', size=20)
17
  # rc('text', usetex=True)
18
- fig, ax = plt.subplots()
19
  axs = ax.twinx()
20
 
21
  def Bernoulli(theta, N, h):
@@ -31,11 +31,11 @@ axs.text(h/N,2,'MLE', color='b')
31
  axs.plot(theta, [scipy.stats.beta.pdf(t, h+alpha, N-h+beta) for t in theta], color='r')
32
  ax.text(theta[0], Likelihood[0],'Likelihood', color='b')
33
  axs.text(theta[-5], scipy.stats.beta.pdf(theta, alpha,beta)[-5],'Prior')
34
- axs.text(alpha/(alpha+beta)-0.1,1,'Prior mean')
35
  axs.text(theta[0],scipy.stats.beta.pdf(theta[0], h+alpha, N-h+beta),'Posterior',color='r')
36
- axs.text((h+alpha)/(N+alpha+beta)-0.1,3,'Post. Mean',color='r')
37
- ax.vlines(alpha/(alpha+beta), *ax.get_ylim(), linestyle='--',label='Prior mean',color='k')
38
- ax.vlines((h+alpha)/(N+alpha+beta), *ax.get_ylim(), linestyle='--',label='Post. Mean',color='r')
39
  # ax.set_title(f"n_samples={int(N)}, n_heads={int(h)}");
40
  ax.tick_params(axis='y', colors='b')
41
  axs.tick_params(axis='y', colors='r')
 
15
  theta = np.linspace(0.01,0.99,100)
16
  #rc('font', size=20)
17
  # rc('text', usetex=True)
18
+ fig, ax = plt.subplots(figsize=(10,5))
19
  axs = ax.twinx()
20
 
21
  def Bernoulli(theta, N, h):
 
31
  axs.plot(theta, [scipy.stats.beta.pdf(t, h+alpha, N-h+beta) for t in theta], color='r')
32
  ax.text(theta[0], Likelihood[0],'Likelihood', color='b')
33
  axs.text(theta[-5], scipy.stats.beta.pdf(theta, alpha,beta)[-5],'Prior')
34
+ # axs.text(alpha/(alpha+beta)-0.1,1,'Prior mean')
35
  axs.text(theta[0],scipy.stats.beta.pdf(theta[0], h+alpha, N-h+beta),'Posterior',color='r')
36
+ # axs.text((h+alpha)/(N+alpha+beta)-0.1,3,'Post. Mean',color='r')
37
+ # ax.vlines(alpha/(alpha+beta), *ax.get_ylim(), linestyle='--',label='Prior mean',color='k')
38
+ # ax.vlines((h+alpha)/(N+alpha+beta), *ax.get_ylim(), linestyle='--',label='Post. Mean',color='r')
39
  # ax.set_title(f"n_samples={int(N)}, n_heads={int(h)}");
40
  ax.tick_params(axis='y', colors='b')
41
  axs.tick_params(axis='y', colors='r')