Several notebook cells perform critical operations (data splitting, model definition, training), but lack in-line comments or markdown explanations. Add concise explanations in markdown above each major block to improve readability.