search

dreamquark-ai / tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
https://dreamquark-ai.github.io/tabnet/
MIT License
2.56k stars 473 forks source link

add f1_score metrics #429

Closed Timaos123 closed 1 year ago

Timaos123 commented 1 year ago

Hi, I added f1_score as a metric in the forked project. If you need to add this function, call it, please. Timaos123/tabnet

chrislemke commented 1 year ago

Additionally average_precision_score would be great.

Optimox commented 1 year ago

@Timaos123 @chrislemke Isn't it easy enough to specify a custom metric ?

I don't think I'll try to implement every metric someone is willing to use.