Inspired by Amazon SageMaker and Ray, try to refactor the classes for hyperparameters. e.g. - Categorical - Continuous - Logarithmic: for small scale, e.g. learning rate