intersample_attention_heads: Get intersample attention heads

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

Description

Get intersample attention heads

Usage

intersample_attention_heads(model, dataset, n = 100, device = "cpu")

Arguments

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

Value

average intersample attention heads as a matrix

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