simple_nn_foward_prop.py

import numpy as np

def forward_prop(X, params):

  """Performs forward propagation and calculates output value

    Arguments
    ---------
    X: array_like
      Data
    params: dictionary
      Parameter dict contaning 'w' and 'b'

    Returns
    -------
    dict
      Dictionary contaning 'z' and 'a'
  """

  w = params['w']
  b = params['b']

  z = np.dot(X, w) + b
  a = sigmoid(z)
  
  return {'z': z, 'a': a}