Tensorflow.js有一个所有标识符的列表

问题描述 投票:0回答:2

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

提前谢谢!

javascript tensorflow identifier tensorflow.js
2个回答
0
投票

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

sebastian-speitel的答案也是正确的,TF.js API docs中显示的类/函数名与字符串表示之间通常有1:1的对应关系。

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

我们当然应该在TF.js API文档中更好地记录这一点。在https://github.com/tensorflow/tfjs/issues/1315跟踪。


0
投票

在官方文档中的losses下,您可以找到所有可能的损失函数。其他的一样。

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

© www.soinside.com 2019 - 2024. All rights reserved.