获取嵌入大小#

pytorch_forecasting.utils._utils.get_embedding_size(n: int, max_size: int = 100) int[来源]#

从经验上确定好的嵌入尺寸(公式来自fastai)。

Parameters:
  • n (int) – 类的数量

  • max_size (int, 可选) – 最大嵌入大小。默认为100。

Returns:

嵌入大小

Return type:

整数