如何为randomsearchCV (永远运行)分配时间限制？

from sklearn.model_selection import RandomizedSearchCV
from scipy.optimize import OptimizeResult
import time

def time_limit_callback(estimator, remaining_time):
    if remaining_time <= 0:
        raise KeyboardInterrupt

class TimeLimitedRandomizedSearchCV(RandomizedSearchCV):
    def fit(self, X, y=None, groups=None, **fit_params):
        self._start_time = time.time()
        self._max_time = self.param_distributions['max_time']
        self._callback = self.param_distributions['callback']
        self._callback(self, self._max_time)
        super().fit(X, y, groups, **fit_params)

param_space = {
    'param1': [value1, value2, ...],
    'param2': [value1, value2, ...],
    ...
    'max_time': [time_limit],
    'callback': [time_limit_callback]
}

random_search = TimeLimitedRandomizedSearchCV(estimator, param_space, n_iter=100, cv=5)