pytorch_forecasting.utils.get_embedding_size

pytorch_forecasting.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

pytorch_forecasting.utils.create_mask pytorch_forecasting.utils.groupby_apply