Case study: fitting classifier models in pyspark