Utils

AutoUMAP

init

add_auto_umap_args

on_train_start

plot

on_validation_end

OfflineUMAP

init

plot

Checkpointer

init

add_checkpointer_args

initial_setup

save_args

save

on_train_start

on_validation_end

Gather layer

Weighted KNN Classifier

init

update

compute

LARS

Metrics

accuracy_at_k

weighted_mean

Momentum module

MomentumUpdater

initialize_momentum_params

Sinkhorn-Knopp

Whitening