top_k_accuracy: Top_k_accuracy

In fastai: Interface to 'fastai'

Top_k_accuracy

Description

Computes the Top-k accuracy ('targ' is in the top 'k' predictions of 'inp')

Usage

top_k_accuracy(inp, targ, k = 5, axis = -1)

Arguments

inp	predictions
targ	targets
k	k
axis	axis

Value

None

Examples

## Not run: 

loaders = loaders()

data = Data_Loaders(loaders['train'], loaders['valid'])$cuda()

model = nn$Sequential() +
  nn$Flatten() +
  nn$Linear(28L * 28L, 10L)
metrics = list(accuracy,top_k_accuracy)
learn = Learner(data, model, loss_func = F$cross_entropy, opt_func = Adam,
                metrics = metrics)


## End(Not run)

fastai documentation built on June 22, 2024, 11:15 a.m.