Skip to content
Snippets Groups Projects
  • sethah's avatar
    b11887c0
    [SPARK-14264][PYSPARK][ML] Add feature importance for GBTs in pyspark · b11887c0
    sethah authored
    ## What changes were proposed in this pull request?
    
    Feature importances are exposed in the python API for GBTs.
    
    Other changes:
    * Update the random forest feature importance documentation to not repeat decision tree docstring and instead place a reference to it.
    
    ## How was this patch tested?
    
    Python doc tests were updated to validate GBT feature importance.
    
    Author: sethah <seth.hendrickson16@gmail.com>
    
    Closes #12056 from sethah/Pyspark_GBT_feature_importance.
    b11887c0
    History
    [SPARK-14264][PYSPARK][ML] Add feature importance for GBTs in pyspark
    sethah authored
    ## What changes were proposed in this pull request?
    
    Feature importances are exposed in the python API for GBTs.
    
    Other changes:
    * Update the random forest feature importance documentation to not repeat decision tree docstring and instead place a reference to it.
    
    ## How was this patch tested?
    
    Python doc tests were updated to validate GBT feature importance.
    
    Author: sethah <seth.hendrickson16@gmail.com>
    
    Closes #12056 from sethah/Pyspark_GBT_feature_importance.