• Belinda Trotta's avatar
    Top k multi error (#2178) · b3db9e92
    Belinda Trotta authored
    * Implement top-k multiclass error metric. Add new parameter top_k_threshold.
    
    * Add test for multiclass metrics
    
    * Make test less sensitive to avoid floating-point issues.
    
    * Change tabs to spaces.
    
    * Fix problem with test in Python 2. Refactor to use np.testing. Decrease number of training rounds so loss is larger and easier to compare.
    
    * Move multiclass tests into test_engine.py
    
    * Change parameter name from top_k_threshold to multi_error_top_k.
    
    * Fix top-k error metric to handle case where scores are equal. Update tests and docs.
    
    * Change name of top-k metric to multi_error@k.
    
    * Change tabs to spaces.
    
    * Fix formatting.
    
    * Fix minor issues in docs.
    b3db9e92
test_engine.py 67.5 KB