attention_heads: Get attention heads
In cmcmaster1/torchtabular: Package for training tabular models using the {torch} package.

Get attention heads

1	attention_heads(model, dataset, n = 100, device = "cpu")

`model`	the tabtransformer model.
`dataset`	dataset to pass through the model to generate attention heads.
`n`	number of rows to use from the dataset. Default is 100.
`device`	whether to run this on 'cpu' or 'cuda'

average attention heads as a matrix

cmcmaster1/torchtabular documentation built on Dec. 19, 2021, 5:17 p.m.

cmcmaster1/torchtabular index

Note that we can't provide technical support on individual packages. You should contact the package authors for that.

Description