CategoryEncodingΒΆ

A preprocessing layer which encodes integer features.

Abstract Signature:

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

PyTorch

API: torch.nn.functional.one_hot
Strategy: Direct Mapping

JAX (Core)

API: jax.nn.one_hot
Strategy: Direct Mapping

Keras

API: keras.layers.CategoryEncoding
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.CategoryEncoding
Strategy: Direct Mapping

Apple MLX

API: β€”
Strategy: Custom / Partial

Flax NNX

API: β€”
Strategy: Custom / Partial

PaxML / Praxis

API: β€”
Strategy: Custom / Partial