Integrate with Optuna

[pplonski]

Maybe it is a good idea to integrate the parameters tuning part with the Optuna framework. The training for sure will take a long time, but the results might be very good.