0
点赞
收藏
分享

微信扫一扫

Python:实现Sigmoid函数算法(附完整源码)


Python:实现Sigmoid函数算法

import numpy as np


def sigmoid(vector: np.array) -> np.array:
"""
Implements the sigmoid function

Parameters:
vector (np.array): A numpy array of shape (1,n)
consisting of real values

Returns:
sigmoid_vec (np.array): The input numpy array, after applying
sigmoid.

Examples:
>>> sigmoid(np.array([-1.0, 1.0, 2.0]))
array([0.26894142, 0.73105858, 0.88079708])

>>> sigmoid(np.array([0.0]))
array([0.5])
"""
return 1 / (1 + np.exp(-vector))


if __name__ == "__main__":
import doctest

doctest.testmod()


举报

相关推荐

0 条评论