(WIP) Hyper-parameter optimization for sklearn