• ranchlai's avatar
    Add multi-label text classification support to pytorch example (#24770) · 8fd8c8e4
    ranchlai authored
    * Add text classification example
    
    * set the problem type and finetuning task
    
    * ruff reformated
    
    * fix bug for unseting label_to_id for regression
    
    * update README.md
    
    * fixed finetuning task
    
    * update comment
    
    * check if label exists in feature before removing
    
    * add useful logging
    8fd8c8e4
run_classification.py 31.8 KB