Label Encoding In Pyspark