view tvii/sigmoid.py @ 92:f1d1f2388fd6

test linear regression
author Jeff Hammel <k0scist@gmail.com>
date Sun, 17 Dec 2017 14:26:15 -0800
parents 38aa9098bf2d
children
line wrap: on
line source

"""
sigmoid function: 1/(1 + e^-z)
"""

import numpy as np
from .iterable import isiterable

def sigmoid(z):
    """https://en.wikipedia.org/wiki/Sigmoid_function"""
    if not isinstance(z, np.ndarray) and isiterable(z):
        z = np.array(z)
    return 1./(1. + np.exp(-z))


def sigmoidprime(z):
    """"derivative of sigmoid"""
    return sigmoid(z)*(1-sigmoid(z))