add support for inhibition in the streamlit demo

emdgroup · May 26, 2021 · cced74c · cced74c
1 parent a745bf6
commit cced74c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/demos/demo_image.py b/demos/demo_image.py
@@ -20,6 +20,7 @@
 n_atoms = st.sidebar.number_input('Number of atoms', min_value=1, value=10)
 atom_size = st.sidebar.number_input('Atom size', min_value=1, value=10)
 sparsity_H = st.sidebar.number_input('Activation sparsity', min_value=0., value=0., step=0.1)
+inhibition = st.sidebar.number_input('Activation lateral inhibition', min_value=0., value=0., step=0.1)
 n_iterations = st.sidebar.number_input('Number of iterations', min_value=1, value=100)
 
 img = racoon_image(gray=not color, scale=scale)
@@ -35,6 +36,7 @@
 nmf.fit(
     V,
     sparsity_H=sparsity_H,
+    inhibition_strength=inhibition,
     progress_callback=lambda _, x: progress_bar.progress((x + 1) / n_iterations)
 )