TensorFlow.js 训练使用多输入一个输出

大佬教程收集整理的这篇文章主要介绍了TensorFlow.js 训练使用多输入一个输出，大佬教程大佬觉得挺不错的，现在分享给大家，也给大家做个参考。

如何解决TensorFlow.js 训练使用多输入一个输出？

开发过程中遇到TensorFlow.js 训练使用多输入一个输出的问题如何解决？下面主要结合日常开发的经验，给出你关于TensorFlow.js 训练使用多输入一个输出的解决方法建议，希望对你解决TensorFlow.js 训练使用多输入一个输出有所启发或帮助；

我目前正在尝试编写一个可以将特定数字序列分类为动作的系统。 到目前为止，尝试使用 tensorflow.js 构建它运行良好，但现在我遇到了一些问题。

我正在尝试使用像这样的输入来训练模型

[
    [0,1,2,3,4,5,6,7,8,9],[0,]

例如应该导致输出 [1]

制作张量如下

let t = tf.tensor2d([...Array.from({ length: (5 * 25) },(v,i) => i)],[5,25])

//Tensor
//   [[0,...,22,23,24 ],//     [25,26,27,47,48,49 ],//     [50,51,52,72,73,74 ],//     [75,76,77,97,98,99 ],//     [100,101,102,122,123,124]]

现在这看起来像我想要的，现在 t.shape 给了我 [ 5,25 ]，我认为我可以将其用作模型的 inputShape，经过研究，我发现我需要将其定义为 {{1 }} 而不是 batchinputShape，这又消除了一个错误。

inputShape

现在我遇到了模型将张量视为多个值的问题，因此寻找多个输出，但我试图让它只给出一个输出

运行时 loadedModel.add(tf.layers.dense({ units: 256,activation: 'relu',batchinputShape: [5,25] }));

我明白了 await loadedModel.fit(t,tf.tensor([0]),{ epochs: 5 })

我认为这表明了我上面写的内容。

现在我的最后一个问题是，我应该如何设置 inputShape（如果那是我做错了），以便只需要 一个 输出。

亲切的问候

解决方法

大佬总结

以上是大佬教程为你收集整理的TensorFlow.js 训练使用多输入一个输出全部内容，希望文章能够帮你解决TensorFlow.js 训练使用多输入一个输出所遇到的程序开发问题。

如果觉得大佬教程网站内容还不错，欢迎将大佬教程推荐给程序员好友。

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ：384754419，请注明来意。