View source on GitHub
|
Kronecker-Factored Lattice layer with monotonicity constraints.
Keras implementation of tensorflow Kronecker-Factored Lattice layer. This layer takes one or more d-dimensional input(s) and combines them using a Kronecker-Factored Lattice function, satisfying monotonicity constraints if specified.
Classes
class BiasInitializer: Initializes bias depending on output_min and output_max.
class KFLRandomMonotonicInitializer: Initializes a tfl.layers.KroneckerFactoredLattice as random monotonic.
class KroneckerFactoredLatticeConstraints: Constraints for tfl.layers.KroneckerFactoredLattice layer.
class ScaleConstraints: Constraints for tfl.layers.KroneckerFactoredLattice scale.
class ScaleInitializer: Initializes scale depending on output_min and output_max.
Functions
create_kernel_initializer(...): Returns a kernel Keras initializer object from its id.
create_scale_initializer(...): Returns a scale Keras initializer object from its id.
View source on GitHub