BigML-CS-UCLA / SpuCo

SpuCo is a Python package developed to further research to address spurious correlations.
https://spuco.readthedocs.io/en/latest/
MIT License
21 stars 7 forks source link

Trainer with early-stopping based on F1 #18

Closed YuYang0901 closed 1 year ago

YuYang0901 commented 1 year ago

Use this trainer instead of the standard one to use the validation data to decide when to make group inferences for different classes with SPARE. It saves one reference model for each class when the average F1 score of the inferred groups is the highest.