Skip to content

Commit

Permalink
Make layer serializable
Browse files Browse the repository at this point in the history
  • Loading branch information
louisPoulain committed Sep 4, 2024
1 parent 38fe849 commit 69ea307
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlpp_lib/probabilistic_layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -832,7 +832,7 @@ def output(self):
"""This allows the use of this layer with the shap package."""
return super(IndependentTruncatedNormal, self).output[0]


@tf.keras.saving.register_keras_serializable()
class IndependentSoftClipCensored(tfpl.DistributionLambda):
"""An independent censored normal Keras layer."""

Expand Down

0 comments on commit 69ea307

Please sign in to comment.