当前位置: 代码迷 >> JavaScript >> Tensorflow.js 是否有所有标识符的列表
  详细解决方案

Tensorflow.js 是否有所有标识符的列表

热度:55   发布时间:2023-06-07 15:25:14.0

我对 Tensorflow 有点陌生,我开始在一个项目中使用 Tensorflow.js,但我遇到了一些问题:在许多领域(sa 激活、优化器、损失、指标等)我们可以使用字符串引用适合我们需要的对象,对于某些我们可以使用直接对象引用(sa tf.losses.sigmoidCrossEntropy),我发现这个版本更容易,因为我找不到我需要的标识符,这必须存储在某个地方,但是它在哪里 ? 是否有所有可能的标识符的列表? (我搜索了 stack、google 和 quant,但找不到任何东西......)

提前谢谢你!

这不完全是“文档”,显然不是很容易找到,但至少它现在可以解决您的具体问题: :

同样,对于其他字符串标识符(激活、度量、约束、初始化器和正则化器),请参阅下的相邻文件。

一般来说,这些应该匹配用于 Python Keras 的字符串标识符。 例如, 显示name='categorical_crossentropy' 然而,TFJS 的实现可能不是 Python Keras 的最新实现(特别是,过去一两年在 Python 方面添加的任何新损失等可能尚未在 TFJS 中实现)。

Tensorflow.js 层遵循 Keras API,因此您可以在的文档中找到所有字符串标识符。 例如, //keras.io/activations/ 列出了“softmax”、“elu”等。

sebastian-speitel 的回答也是正确的, 显示的类/函数名称与字符串表示之间通常存在 1:1 的对应关系。

一个问题:Keras 使用snake_case 字符串标识符,例如hard_sigmoid ,但在TF.js 中我们需要lowerCamelCase,例如hardSigmoid (大多数标识符都是一个词,在这种情况下没有区别)。

我们当然应该在 TF.js API 文档中更好地记录这一点。 在跟踪。

在的正式文件中,你可以找到所有可能的损失函数。 其他的也一样。

我不是 100% 确定,但它们中的每一个都应该能够用作字符串引用。

  相关解决方案