Add feature importance plot for model based selection and remove button

app.py

@@ -16,14 +16,18 @@ def select_features(method,num_features):
     feature_names = np.array(diabetes.feature_names)
     if method == 'model':
         importance = np.abs(ridge.coef_)
-        threshold = np.sort(importance)[-3] + 0.01
         tic = time()
-        sfm = SelectFromModel(ridge, threshold=threshold).fit(X, y)
+        sfm = SelectFromModel(ridge, threshold=-np.inf,max_features=num_features).fit(X, y)
         toc = time()
         selected_features = feature_names[sfm.get_support()]
         if int(num_features) < len(selected_features):
             selected_features = selected_features[:int(num_features)]
         execution_time = toc - tic
+        fig, ax = plt.subplots()
+        ax.bar(height=importance, x=feature_names)
+        ax.set_title("Feature importances via coefficients")
+        ax.set_ylabel("Importance coefficient")
+        ax.set_xlabel("Features")
     elif method == 'sfs-forward':
         tic_fwd = time()
         sfs_forward = SequentialFeatureSelector(
@@ -32,6 +36,8 @@ def select_features(method,num_features):
         toc_fwd = time()
         selected_features = feature_names[sfs_forward.get_support()]
         execution_time = toc_fwd - tic_fwd
+        importance = np.abs(sfs_forward.get_params()['estimator'].coef_)        
+        fig = None
     elif method == 'sfs-backward':
         tic_bwd = time()
         sfs_backward = SequentialFeatureSelector(
@@ -40,7 +46,10 @@ def select_features(method,num_features):
         toc_bwd = time()
         selected_features = feature_names[sfs_backward.get_support()]
         execution_time = toc_bwd - tic_bwd
-    return f"Selected the following features: {', '.join(selected_features)} in {execution_time:.3f} seconds"
+        importance = np.abs(sfs_backward.get_params()['estimator'].coef_)
+        fig = None
+
+    return f"Selected the following features: {', '.join(selected_features)} in {execution_time:.3f} seconds", fig
 
 title = "Selecting features with Sequential Feature Selection"
 with gr.Blocks(title=title) as demo:
@@ -68,9 +77,9 @@ with gr.Blocks(title=title) as demo:
 
     method = gr.Radio(["model", "sfs-forward", "sfs-backward"], label="Method")
     num_features = gr.Slider(minimum=2, maximum=10, step=1, label = "Number of features")
-    output = gr.Textbox(label="Output Box")
-    select_btn = gr.Button("Select")
-    select_btn.click(fn=select_features, inputs=[method,num_features], outputs=output)
+    output = gr.Textbox(label="Selected features")
+    plot = gr.Plot(label="Feature importance plot")
+    num_features.change(fn=select_features, inputs=[method,num_features], outputs=[output,plot])
 
 demo.launch()