Numpy-Neuron / numpyneuron /activation.py
Jensen Holm
adding unit tests for activation functions (using pytest)
ff1254a
raw
history blame
1.24 kB
import numpy as np
from abc import abstractmethod, ABC
class Activation(ABC):
@abstractmethod
def forward(self, X: np.ndarray) -> np.ndarray:
pass
@abstractmethod
def backward(self, X: np.ndarray) -> np.ndarray:
pass
class Relu(Activation):
def forward(self, X: np.ndarray) -> np.ndarray:
return np.maximum(0, X)
def backward(self, X: np.ndarray) -> np.ndarray:
return np.where(X > 0, 1, 0)
class TanH(Activation):
def forward(self, X: np.ndarray) -> np.ndarray:
return np.tanh(X)
def backward(self, X: np.ndarray) -> np.ndarray:
return 1 - self.forward(X) ** 2
class Sigmoid(Activation):
def forward(self, X: np.ndarray) -> np.ndarray:
return 1.0 / (1.0 + np.exp(-X))
def backward(self, X: np.ndarray) -> np.ndarray:
s = self.forward(X)
return s - (1 - s)
class SoftMax(Activation):
def forward(self, X: np.ndarray) -> np.ndarray:
ax = 1 if X.ndim > 1 else 0
exps = np.exp(
X - np.max(X, axis=ax, keepdims=True)
) # Avoid numerical instability
return exps / np.sum(exps, axis=ax, keepdims=True)
def backward(self, X: np.ndarray) -> np.ndarray:
return X