https://machinelearningmodels.org/bert-machine-learning-model-reshaping-nlp/