예제 #1
0
파일: widget.py 프로젝트: xcliufb/captum
 def _fetch_data(self, change):
     if not self.config:
         return
     with self.out:
         self.visualizer._update_config(self.config)
         self.output = namedtuple_to_dict(self.visualizer.visualize())
         self.config = dict()
예제 #2
0
 def get_insights_config(self):
     return {
         "classes": self.classes,
         "methods": list(ATTRIBUTION_NAMES_TO_METHODS.keys()),
         "method_arguments": namedtuple_to_dict(ATTRIBUTION_METHOD_CONFIG),
         "selected_method": self._config.attribution_method,
     }
예제 #3
0
파일: widget.py 프로젝트: orionr/captum-2
 def _fetch_attribution(self, change):
     if self.label_details:
         self.attribution = namedtuple_to_dict(
             self.visualizer._calculate_attribution_from_cache(
                 self.label_details["instance"],
                 self.label_details["labelIndex"]))
         self.label_details = dict()