fix key errors

dianna-ai · Oct 21, 2024 · c580c1a · c580c1a
1 parent a6d1dce
commit c580c1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dianna/dashboard/_models_text.py b/dianna/dashboard/_models_text.py
@@ -27,7 +27,8 @@ def _run_rise_text(_model, text, **kwargs):
 
 @st.cache_data
 def _run_lime_text(_model, text, **kwargs):
-    relevances = explain_text(_model, text, tokenizer, method='LIME', **kwargs)
+    relevances = explain_text(_model, text, tokenizer, method='LIME',
+                              **kwargs)
     return relevances
 
 

diff --git a/dianna/dashboard/_shared.py b/dianna/dashboard/_shared.py
@@ -117,7 +117,7 @@ def _get_params(method: str, key):
         else:
             return {
                 'random_state': st.number_input('Random state', value=2, key=f'{key}_{method}_rs'),
-                'num_features': 999,
+                'num_features': st.number_input('Number of features', 999, key=f'{key}_{method}_rf'),
                 'num_samples': st.number_input('Number of samples', value=2000, key=f'{key}_{method}_ns')
             }