relu activation

This commit is contained in:
trian-gles 2024-06-28 09:13:47 +02:00
parent 77f8d3bf57
commit 2e22f60c2e

2
tf.js
View File

@ -12,7 +12,7 @@ var model = tf.sequential();
model.add(tf.layers.dense({units: hiddenSize, inputShape: [inputShape]}));
model.add(tf.layers.dense({units: hiddenSize, inputShape: [inputShape], activation: 'relu'}));
model.add(tf.layers.dense({units: outputShape}));