Nipun commited on
Commit
ba5f5b3
·
unverified ·
2 Parent(s): f295ac56952466

Merge pull request #1 from nipunbatra/feature/add-beta-distribution

Browse files
Files changed (1) hide show
  1. config/continuous/beta.py +11 -0
config/continuous/beta.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import torch
3
+
4
+ BETA = {
5
+ "params": lambda: {
6
+ "alpha": st.sidebar.slider("Alpha (α)", 0.1, 10.0, 2.0, 0.1),
7
+ "beta": st.sidebar.slider("Beta (β)", 0.1, 10.0, 2.0, 0.1),
8
+ },
9
+ "dist": lambda p: torch.distributions.Beta(p["alpha"], p["beta"]),
10
+ "support": lambda p: (0, 1),
11
+ }