问题标签 [tensorflow-layers]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
1 回答
144 浏览

javascript - 将 keras 模型加载到 tfjs 会导致输入形状不匹配

我在 Keras 中做了一个模型并将其保存为 tfjs 模型。我成功地将它导入到我的 js 项目中。但是,由于输入形状错误,我无法在我的 JS 代码中使用该模型。凯拉斯模型:

当我在这样的 python 文件中使用它时,它可以按预期工作,例如:

但是,当我在 JS 中尝试相同的

给出这个错误:

tfjs@latest:17 Uncaught (in promise) 错误:检查时出错:预期dense_13_input 的形状为[null,4],但得到的数组的形状为[4,1]。

我不断收到相同的错误:

  1. 将 s1v...s4v 单独声明为 tf.scalars 并使用 tf.stack() 堆叠它们
  2. 将 s1v...s4v 单独声明为 tf.tensor1d 并使用 tf.stack() 堆叠它们
  3. 在单个 tf.tensor1d 中声明 s1...s4v

请帮我解决这个问题...

0 投票
0 回答
24 浏览

tensorflow - 来自多输入模型拱门的 Logits。从熵损失函数到单个 Tensorflow sigmoid

我已经实现了绝大多数的多输入模型,并且我在为 2 个传入的 logit 张量定义单个损失函数的最后几个步骤中苦苦挣扎。

可以说我有以下内容:

然后我想将它们提供给以下损失函数:

我在想这样的事情:

损失实现有意义吗?我觉得这不是一个概念上正确的方法。例如,如果yis shape[64,1]logits_concatis shape [64,2],那么我建议有 2 个类在起作用,这不是我的意图。