mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-23 21:11:08 +00:00
Add binary step activation function (#10030)
* Add binary step activation function * fix: ruff line too long error * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * refactor: add link to directory * revert: add link to directory * fix: algorithm bug and docs * Update neural_network/activation_functions/binary_step.py * fix: ruff line too long error --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Tianyi Zheng <tianyizheng02@gmail.com>
This commit is contained in:
parent
2260961a80
commit
ed19b1cf0c
36
neural_network/activation_functions/binary_step.py
Normal file
36
neural_network/activation_functions/binary_step.py
Normal file
|
@ -0,0 +1,36 @@
|
|||
"""
|
||||
This script demonstrates the implementation of the Binary Step function.
|
||||
|
||||
It's an activation function in which the neuron is activated if the input is positive
|
||||
or 0, else it is deactivated
|
||||
|
||||
It's a simple activation function which is mentioned in this wikipedia article:
|
||||
https://en.wikipedia.org/wiki/Activation_function
|
||||
"""
|
||||
|
||||
|
||||
import numpy as np
|
||||
|
||||
|
||||
def binary_step(vector: np.ndarray) -> np.ndarray:
|
||||
"""
|
||||
Implements the binary step function
|
||||
|
||||
Parameters:
|
||||
vector (ndarray): A vector that consists of numeric values
|
||||
|
||||
Returns:
|
||||
vector (ndarray): Input vector after applying binary step function
|
||||
|
||||
>>> vector = np.array([-1.2, 0, 2, 1.45, -3.7, 0.3])
|
||||
>>> binary_step(vector)
|
||||
array([0, 1, 1, 1, 0, 1])
|
||||
"""
|
||||
|
||||
return np.where(vector >= 0, 1, 0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import doctest
|
||||
|
||||
doctest.testmod()
|
Loading…
Reference in New Issue
Block a user