Skip to content

Commit 9b92475

Browse files
committed
fix(KDP): added a missing code for the example for disttribution aware layer for custom pipelines
1 parent dd728ef commit 9b92475

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

docs/example_usages.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -298,6 +298,8 @@ However we can also manually set the prefered distribution for each numerical fe
298298
```python
299299
from kdp.features import NumericalFeature, FeatureType
300300
from kdp.processor import PreprocessingModel, OutputModeOptions
301+
from kdp.custom_layers import DistributionAwareEncoder
302+
301303

302304
# Define features
303305
features = {
@@ -321,6 +323,7 @@ features = {
321323
preprocessors=[
322324
tf.keras.layers.Rescaling,
323325
tf.keras.layers.Normalization,
326+
DistributionAwareEncoder,
324327
],
325328
bin_boundaries=[0.0, 1.0, 2.0],
326329
mean=0.0,
8.97 KB
Loading

0 commit comments

Comments
 (0)