simple_nn_initialization.py

import numpy as np

def initialize_nn(X):

  """Initializes random weights and bias

  Arguments
  ---------
  X: array-like
    Train Dataset

  Returns
  -------
  dict
    Contains the randomly initialized weights and bias arrays where-
    shape(weights) = (X.Shape[0], 1)
    bias = float value

    The keys for weights and bias arrays in the dict is 'w' and 'b'
  """

  np.random.seed(999)

  w = np.random.randn(X.shape[1], 1) * 0.01
  b = 0

  return {'w': w, 'b': b}