Update script.js

script.js

@@ -143,7 +143,7 @@ var dojob = function(run_id) {
                 tf.dispose(gru0_hx_);
                 const att_inp = tf.concat([att_context, input, out_0], 1);
                 const att_params = tf.add(tf.matMul(att_inp, att_w), att_b);
-                [alpha,beta,kappa] = tf.split(tf.softplus(att_params), 3, 1);
+                [alpha,beta,kappa] = tf.split(tf.log(tf.add(tf.exp(att_params), 1.)), 3, 1);
                 att_location_ = att_location;
                 att_location = tf.add(att_location, tf.div(kappa, 25.));
                 tf.dispose(att_location_)