Error ParallelMapDatasetV2 when trying to predict a model using Keras
I’m trying to train a simple classification model using Keras that works with text and has two output classification: 0 or 1. I transformed my text data using TfidVectorizer() from SKlearn and everything ran well, until I tried to predict using the Test dataset (here called X_test_tfid). I thought it could be because this is a sparse matrix and Keras could not handle it very well, so I even transformed to a TensorFlowSparse using a very simple transformation algorithm, but the same error occurs. I have no idea what could be possible happening. I appreciate any advices.
Notice that if I try to “predict” using the training dataset everything runs just fine. I will copy the main parts of the code here since it’s a simple code: