https://jmtirado.net/posts/20240124_pyspark_mllib/