• Ilya Matiach's avatar
    adding sparse support to TreeSHAP in lightgbm (#3000) · 9f367d11
    Ilya Matiach authored
    * adding sparse support to TreeSHAP in lightgbm
    
    * updating based on comments
    
    * updated based on comments, used fromiter instead of frombuffer
    
    * updated based on comments
    
    * fixed limits import order
    
    * fix sparse feature contribs to work with more than int32 max rows
    
    * really fixed int64 max error and build warnings
    
    * added sparse test with >int32 max rows
    
    * fixed python side reshape check on sparse data
    
    * updated based on latest comments
    
    * fixed comments
    
    * added CSC INT32_MAX validation to test, fixed comments
    9f367d11
test_engine.py 106 KB