From 1b3bf75faec3cec8db61c62bc365f25ed054bd3f Mon Sep 17 00:00:00 2001
From: neelbhandari6 <30409992+neelbhandari6@users.noreply.github.com>
Date: Sun, 19 Jul 2020 10:22:15 +0530
Subject: [PATCH] Update model.py

---
 src/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index 230b83cc2..e4eb138c0 100644
--- a/src/model.py
+++ b/src/model.py
@@ -17,7 +17,7 @@ def shape_list(x):
     dynamic = tf.shape(x)
     return [dynamic[i] if s is None else s for i, s in enumerate(static)]
 
-def softmax(x, axis=-1):
+def softmax(x, axis=-1):"""Normalises the probability function to prevent the overflow of all elements"""
     x = x - tf.reduce_max(x, axis=axis, keepdims=True)
     ex = tf.exp(x)
     return ex / tf.reduce_sum(ex, axis=axis, keepdims=True)