get_embedding_size

pytorch_forecasting.utils._utils.get_embedding_size(n: int, max_size: int = 100) → int

Determine empirically good embedding sizes (formula taken from fastai).

Parameters:

• n (int) – Number of classes.

• max_size (int, optional) – Maximum embedding size. Defaults to 100.

Returns:

Embedding size.

Return type:

int