36 lines
622 B
Python
36 lines
622 B
Python
import numpy as np
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
class ActivationFunc(ABC):
|
|
@abstractmethod
|
|
def d(v: np.ndarray) -> np.ndarray:
|
|
pass
|
|
|
|
|
|
class ReLU(ActivationFunc):
|
|
def __call__(self, x):
|
|
return x * (x > 0)
|
|
|
|
def d(self, x):
|
|
return x > 0
|
|
|
|
|
|
class LeakyReLU(ActivationFunc):
|
|
def __init__(self, k=0.01):
|
|
self.k = k
|
|
|
|
def __call__(self, x):
|
|
return x * (x > 0) + self.k * x * (x <= 0)
|
|
|
|
def d(self, x):
|
|
return (x > 0) + self.k * (x <= 0)
|
|
|
|
|
|
class Identity(ActivationFunc):
|
|
def __call__(self, x):
|
|
return x
|
|
|
|
def d(x):
|
|
return 1
|