cbg-ethz / bmi

Mutual information estimators and benchmark
https://cbg-ethz.github.io/bmi/
MIT License
33 stars 5 forks source link

Implement early stopping for neural estimators #73

Closed grfrederic closed 1 year ago

grfrederic commented 1 year ago

The default setup for neural estimators should be training on a 50-50 train/test split, early stopping when test MI stops growing, and return the highest test MI as the estimate.

grfrederic commented 1 year ago

done with commit #81