I have managed to successfully follow the notebook Customizing_embeddings.ipynb . It has improved my similarity classification accuracy significantly. However, I cannot understand exactly how the trained matrix is created. Is the weight matrix the one we get from training the classification model? If you point to the reference in literature to this method, that would be greatly appreciated.