在 TensorflowJS 中构建神经网络
Build neural network in TensorflowJS
我用 nodejs 和 TensorflowJS 构建了一个聊天机器人。
我的代码基于This tutorial
我在 'translate' 神经网络构建方面遇到了麻烦。
发件人:
# Build neural network
net = tflearn.input_data(shape=[None, len(train_x[0])])
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, len(train_y[0]), activation='softmax')
net = tflearn.regression(net)
# Define model and setup tensorboard
model = tflearn.DNN(net, tensorboard_dir='tflearn_logs')
# Start training (apply gradient descent algorithm)
model.fit(train_x, train_y, n_epoch=1000, batch_size=8, show_metric=True)
model.save('model.tflearn')
到目前为止我得到了这个:
// Build neural network:
const model = tf.sequential();
model.add(tf.layers.dense({units: training.length, activation: 'relu', inputShape: [train_x[0].length]}));
model.add(tf.layers.dense({units: train_y[0].length, activation: 'linear'}));
model.compile({optimizer: 'sgd', loss: 'meanSquaredError'});
关键是我不能 'predict' 使用我的 JS 代码。
I got this erro
And this is my xs
完整代码https://github.com/ran-j/ChatBotNodeJS/blob/master/routes/index.js#L184
与 不同,因为它是 python 的 'traduction' 而不是使用创建神经网络
的错误
train_x[0].length = 48
train_y[0].length = 9
我在预测模型时出错了。
试试这个:
var model = tf.sequential();
model.add(tf.layers.dense({
units: 8,
inputShape: [null, train_x[0].length]
}));
model.add(tf.layers.dense({
units: 8
}));
model.add(tf.layers.dense({
units: train_y[0].length
activation: 'softmax'
}));
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
await model.fit(x_tensor, y_tensor, {
epochs: 1000,
batchSize: 8
})
我用 nodejs 和 TensorflowJS 构建了一个聊天机器人。
我的代码基于This tutorial
我在 'translate' 神经网络构建方面遇到了麻烦。
发件人:
# Build neural network
net = tflearn.input_data(shape=[None, len(train_x[0])])
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, len(train_y[0]), activation='softmax')
net = tflearn.regression(net)
# Define model and setup tensorboard
model = tflearn.DNN(net, tensorboard_dir='tflearn_logs')
# Start training (apply gradient descent algorithm)
model.fit(train_x, train_y, n_epoch=1000, batch_size=8, show_metric=True)
model.save('model.tflearn')
到目前为止我得到了这个:
// Build neural network:
const model = tf.sequential();
model.add(tf.layers.dense({units: training.length, activation: 'relu', inputShape: [train_x[0].length]}));
model.add(tf.layers.dense({units: train_y[0].length, activation: 'linear'}));
model.compile({optimizer: 'sgd', loss: 'meanSquaredError'});
关键是我不能 'predict' 使用我的 JS 代码。
I got this erro
And this is my xs
完整代码https://github.com/ran-j/ChatBotNodeJS/blob/master/routes/index.js#L184
与
train_x[0].length = 48
train_y[0].length = 9
我在预测模型时出错了。
试试这个:
var model = tf.sequential();
model.add(tf.layers.dense({
units: 8,
inputShape: [null, train_x[0].length]
}));
model.add(tf.layers.dense({
units: 8
}));
model.add(tf.layers.dense({
units: train_y[0].length
activation: 'softmax'
}));
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
await model.fit(x_tensor, y_tensor, {
epochs: 1000,
batchSize: 8
})