Getting outputs of mode.predict() per sentence input

Models
2

hey @vinurad13 what shape does test_encodings["input_ids"] have?

without seeing the details behind X_test my guess is that you need to reshape your inputs so that input_ids has shape (batch_size, max_seq_length)