Simple softmax implementation

softmax.py

import numpy as np
from math import exp


def softmax(x):
    """Compute softmax values for each sets of scores in x."""
    # S(yi) = e^/yi / sum(ey2)
    y = np.array(x)
    f = np.vectorize(exp)
    n = f(y)
    d = sum(n)
    return n / d