Updating my own grid search since the library does not handle one class correctly.