reconstruct explainer

app.py

@@ -1,6 +1,7 @@
 # python image_gradio.py >> ./logs/image_gradio.log 2>&1
 import time
 import pickle
+import dill
 import os
 import gradio as gr
 import spaces
@@ -543,11 +544,13 @@ class ExplainerCheckbox(Component):
                 # opt_output.explainer.model = self.experiment.model
                 # self.experiment.manager._explainers.append(opt_output.explainer)
                 # self.experiment.manager._explainer_ids.append(opt_exp_id)
-                
+
                 opt_res = {
                     'id': opt_exp_id,
-                    # 'explainer': opt_output.explainer,
+                    'class': opt_output.explainer.__class__,
+                    'params' : opt_output.study.best_trial.params,
                 }
+
                 self.groups.insert_check(checkbox_group_info, self.explainer_name, opt_exp_id, opt_postprocessor_id)
                 checkbox = gr.update(label="Optimized Parameter (Optimal)", interactive=True)
                 bttn = gr.update(value="Optimized", variant="secondary")
@@ -555,9 +558,17 @@ class ExplainerCheckbox(Component):
                 return [opt_res, checkbox_group_info, checkbox, bttn]
 
             def update_exp(exp_res):
-                exp_res['explainer'].model = self.experiment.model
-                # self.experiment.manager._explainers.append(exp_res['explainer'])
-                self.experiment.manager._explainer_ids.append(exp_res['id'])
+                kwargs = {}
+                for k,v in exp_res['params'].items():
+                    if "explainer" in k:
+                        _key = k.split("explainer.")[1]
+                        kwargs[_key] = v
+                
+                kwargs['model'] = self.experiment.model
+                explainer = exp_res['class'](**kwargs)
+                _id = exp_res['id']
+                self.experiment.manager._explainers.append(explainer)
+                self.experiment.manager._explainer_ids.append(_id)
 
             self.bttn.click(optimize, inputs=[self.groups.info], outputs=[self.opt_res, self.groups.info, self.opt_check, self.bttn], queue=True, concurrency_limit=1)
             self.opt_res.change(update_exp, self.opt_res)