Create a capsule class in order to use Optuna lib for hyperparameter optimization.
Create a capsule class in order to use Optuna lib for hyperparameter optimization.