diff --git a/tf.js b/tf.js
index 9d60fcc..d0eabfe 100644
--- a/tf.js
+++ b/tf.js
@@ -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}));