keras-team / keras-io

Keras documentation, hosted live at keras.io
Apache License 2.0
2.75k stars 2.03k forks source link

Set input dtype to int32 for categorical features in tabtransformer example #1815

Closed kuma-guy closed 5 months ago

kuma-guy commented 6 months ago

In create_model_inputs, the same inputs were created for both numeric and categorical variables, so I updated the dtype for categorical variables