Nance-Lab / diff_predictor

Repository for diff_predictor, a software package for Machine Learning predictions of nanoparticle diffusion data
MIT License
1 stars 0 forks source link

Use Optuna to improve hyperparameter optimization times #25

Open nlsschim opened 10 months ago

nlsschim commented 10 months ago

The currently time limiting factor for diff_predictor is the amount of time hyperparameter optimization takes, especially when not using a GPU, which might limit users who dont have GPU access or experience. We could use Optuna: https://optuna.org/#code_examples to make this process faster