View model importance in Spark

model_importance.py

print('Feature Importance:')
    featureImportances = model.stages[7].featureImportances
    for keyPair in spark.sparkContext \
                        .parallelize(zip(features,
                                         featureImportances.toArray())) \
                        .sortBy(lambda pair: -pair[1]) \
                        .collect():
        print('{0}: {1}'.format(keyPair[0], keyPair[1]))