36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from .activation import Activation, act
|
|
from .aggregation import Aggregation, agg
|
|
from .tools import *
|
|
from .graph import *
|
|
|
|
Activation.name2func = {
|
|
'sigmoid': Activation.sigmoid_act,
|
|
'tanh': Activation.tanh_act,
|
|
'sin': Activation.sin_act,
|
|
'gauss': Activation.gauss_act,
|
|
'relu': Activation.relu_act,
|
|
'elu': Activation.elu_act,
|
|
'lelu': Activation.lelu_act,
|
|
'selu': Activation.selu_act,
|
|
'softplus': Activation.softplus_act,
|
|
'identity': Activation.identity_act,
|
|
'clamped': Activation.clamped_act,
|
|
'inv': Activation.inv_act,
|
|
'log': Activation.log_act,
|
|
'exp': Activation.exp_act,
|
|
'abs': Activation.abs_act,
|
|
'hat': Activation.hat_act,
|
|
'square': Activation.square_act,
|
|
'cube': Activation.cube_act,
|
|
}
|
|
|
|
Aggregation.name2func = {
|
|
'sum': Aggregation.sum_agg,
|
|
'product': Aggregation.product_agg,
|
|
'max': Aggregation.max_agg,
|
|
'min': Aggregation.min_agg,
|
|
'maxabs': Aggregation.maxabs_agg,
|
|
'median': Aggregation.median_agg,
|
|
'mean': Aggregation.mean_agg,
|
|
}
|