Nipun's picture
Add beta distribution implementation
6952466
raw
history blame contribute delete
322 Bytes
import streamlit as st
import torch
BETA = {
"params": lambda: {
"alpha": st.sidebar.slider("Alpha (α)", 0.1, 10.0, 2.0, 0.1),
"beta": st.sidebar.slider("Beta (β)", 0.1, 10.0, 2.0, 0.1),
},
"dist": lambda p: torch.distributions.Beta(p["alpha"], p["beta"]),
"support": lambda p: (0, 1),
}