I am working with pyspark. I converted the features VectorAssembler with following code
from pyspark.ml.feature import VectorAssembler assembler = VectorAssem