time-to-botec

Benchmark sampling in different programming languages
Log | Files | Refs | README

samples.py (356B)

      1 import squigglepy as sq
      2 import numpy as np
      3 
      4 p_a = 0.8
      5 p_b = 0.5
      6 p_c = p_a * p_b
      7 
      8 dist_0 = 0
      9 dist_1 = 1
     10 dist_some = sq.to(1, 3)
     11 dist_many = sq.to(2, 10)
     12 
     13 dists = [dist_0, dist_1, dist_some, dist_many]
     14 weights = [(1 - p_c), p_c/2, p_c/4, p_c/4 ]
     15 
     16 result = sq.mixture(dists, weights)
     17 result_samples = sq.sample(result, 1000000)
     18 print(np.mean(result_samples))