Nipun's picture
feat: implement automatic distribution discovery system
f295ac5
raw
history blame contribute delete
255 Bytes
import streamlit as st
import torch
EXPONENTIAL = {
"params": lambda: {
"rate": st.sidebar.slider("Rate (λ)", 0.1, 5.0, 1.0, 0.1),
},
"dist": lambda p: torch.distributions.Exponential(p["rate"]),
"support": lambda p: (0, None),
}