mldas.explore.evaluation.label_2d_latent

mldas.explore.evaluation. label_2d_latent ( model , data_loader , embeddings = False )

Scatter plot of 2D latent space with label-based color schemme.

Parameters

model torch.nn.Module

Trained model.

data_loader torch.utils.data.DataLoader

Input data to evaluate the trained model with.

embeddings bool

Whether to display the embeddings or not.