Update the documentation

• Some training parameters have been removed but are still present in the documentation, we should remove them.
• We need to go over the full documentation to fix typos and some wrong code formatting.
• We should also add missing docstrings.