Skip to content

Commit 37fe0b1

Browse files
teddykokerfacebook-github-bot
authored andcommitted
Correct default method argument for captum.attr.visualization.visualize_t… (#1152)
Summary: The default argument for `method` in `captum.attr.visualization.visualize_timeseries_attr` is currently `"individual_channels"`, which is not a valid option, resulting in an exception if used. This PR changes the default method to `"overlay_individual"`, which is what the docs indicate the default should be. Pull Request resolved: #1152 Reviewed By: aobo-y Differential Revision: D47197267 Pulled By: vivekmig fbshipit-source-id: 4d87f792b742fafbb9c30e84247c830e93df1187
1 parent 645af96 commit 37fe0b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

captum/attr/_utils/visualization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -448,7 +448,7 @@ def visualize_timeseries_attr(
448448
attr: ndarray,
449449
data: ndarray,
450450
x_values: Optional[ndarray] = None,
451-
method: str = "individual_channels",
451+
method: str = "overlay_individual",
452452
sign: str = "absolute_value",
453453
channel_labels: Optional[List[str]] = None,
454454
channels_last: bool = True,

0 commit comments

Comments
 (0)