CategoryEncoding

A preprocessing layer which encodes integer features.

Abstract Signature:

CategoryEncoding(num_tokens: int, output_mode: str = multi_hot, sparse: bool = False)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.nn.functional.one_hot

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.nn.one_hot

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.layers.CategoryEncoding

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.keras.layers.CategoryEncoding

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: —

Strategy: Custom / Partial

PaxML / Praxis

API: —

Strategy: Custom / Partial