Replies: 1 comment 1 reply
-
JAX doesn't include any tools for this kind of text vectorization. I would probably use something like For doing things more at scale, the |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Trying to port a kaggle housing regressor from Tensorflow to Jax but have hit a snag.
The dataset has a few categorical features(like "MSZoning", "Street","Alley") which need to hot encoded. Tensorflow has a few preprocessors like TextVectorization, StringLookup, or IntegerLookup to handle categorical features of various types.
I could build a text category encoder as shown below which would adapt on a particular categorical feature.
And finally plug the encoder in the model as shown below.
Is there anything similar to StringLookup/textvectorizer in jax/flax which could adapt on a category and can be plugged in a model?
Beta Was this translation helpful? Give feedback.
All reactions