time-to-botec

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

samples.js (570B)


      1 import { run } from "@quri/squiggle-lang";
      2 
      3 let squiggle_code = `
      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 = SampleSet.fromDist(1 to 3)
     11 dist_many = SampleSet.fromDist(2 to 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 = mixture(dists, weights)
     17 mean(result)
     18 `
     19 
     20 async function main(){
     21   let output = await run(squiggle_code, {
     22     environment: {
     23       xyPointLength: 1000000,
     24       sampleCount: 1000000,
     25       sparkLine: 20,
     26     }
     27   })
     28   console.log(output.value.result.value)
     29 }
     30 main()