Bert Model wrapper with TensorFlow Wrapper
Id of sentence start Token
Id of sentence end Token.
Configuration for TensorFlow session
labels which model was trained with in order
TF v2 signatures in Spark NLP
Encode the input sequence to indexes IDs adding padding where necessary
Bert Model wrapper with TensorFlow Wrapper