【发布时间】:2019-01-18 06:56:21
【问题描述】:
有人会帮助我完成这个 TensorFlow JS 项目吗? 这是一个具有机器学习功能的聊天机器人,我坚持使用“构建神经网络”, giveme this error
项目链接:https://github.com/ran-j/ChatBotNodeJS
/routes/index.js 第 189 行的训练代码
//Build neural network
model = tf.sequential();
model.add(tf.layers.dense({inputShape: [documents.length], units: 100}));
model.add(tf.layers.dense({units: 4}));
model.compile({loss: 'categoricalCrossentropy', optimizer: 'sgd'});
model.fit(xs, ys, {epochs: 1000});
【问题讨论】:
-
你能给我们看看你的
xs张量吗? -
张量 { isDisposedInternal: false, size: 1296, shape: [ 27, 48 ], dtype: 'float32', strides: [ 48 ], dataId: {}, id: 0, rankType: ' 2' }
-
27是批量输入量还是48还是你的输入是二维的?尝试使用
model.fit(xs.transpose(), ys, {epochs: 1000}); -
我不知道,我的数据是这样的pastebin.com/ktacvqCh
-
那是你的文档数组?
标签: node.js tensorflow.js