rex_xai.extraction ================== .. py:module:: rex_xai.extraction Classes ------- .. autoapisummary:: rex_xai.extraction.Explanation Module Contents --------------- .. py:class:: Explanation(maps, prediction_func, data, args, run_stats, keep_all_maps=False) .. py:attribute:: target_map .. py:attribute:: explanation :type: Optional[torch.Tensor] :value: None .. py:attribute:: final_mask :value: None .. py:attribute:: explanation_confidence :value: 0.0 .. py:attribute:: prediction_func .. py:attribute:: data .. py:attribute:: args .. py:attribute:: run_stats .. py:method:: __repr__() .. py:method:: extract(method) .. py:method:: blank() .. py:method:: set_to_true(coords, mask=None) .. py:method:: __global(map=None, wipe=False) .. py:method:: __generate_circle_coordinates(centre, radius) .. py:method:: __draw_circle(centre, start_radius=None) .. py:method:: compute_masked_responsibility(mask) .. py:method:: __spatial(centre=None, expansion_limit=None) .. py:method:: save(path, mask=None, multi=None, multi_style='', clauses=None) .. py:method:: heatmap_plot(path=None) .. py:method:: surface_plot(path=None) .. py:method:: show(path=None)