enerzyme.models.layers.rbf.BesselRBFLayer

class enerzyme.models.layers.rbf.BesselRBFLayer(num_rbf: int, cutoff_sr: float, cutoff_fn: Literal['polynomial', 'bump'] = 'polynomial', trainable: bool = False)

Bases: BaseRBF

__init__(num_rbf: int, cutoff_sr: float, cutoff_fn: Literal['polynomial', 'bump'] = 'polynomial', trainable: bool = False) → None