LSTM notebook rework

• Add more explanations to LSTM notebook
• Maybe simplify the model used
• give the option to load the model from disk because training takes too long