I am using PySpark (--version 3.0.1) in python and for my train data, I have defined/added a weight column to the DataFrame based on the class label. I have defined the weig