get_embedding_size¶
- pytorch_forecasting.utils.get_embedding_size(n: int, max_size: int = 100) int[source]¶
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