时间戳

首页 > 解决方案 > 限制拟合时间

python - 限制拟合时间

问题描述

我正在使用model_selection.GridSearchCVpython 中的 scikit-learn 库来查找 SVR 的最佳参数并将它们拟合到我的训练集。然而，一些训练集需要很长时间才能适应。我想知道是否有办法限制拟合功能的时间。我现在减少 gamma 和 C 的值会这样做，但我不想降低所有训练集的模型的准确性。这是我正在使用的代码片段：

tuned_parameters = [
    {'kernel':['linear'], 'C':[1,10,100]},
    {'kernel':['rbf'],    'C':[1,10,100], 'gamma': 10.0**np.arange(-3,1)}, 
    {'kernel':['poly'],   'C':[1,10,100], 'gamma': 10.0**np.arange(-3,1),'degree':[2]}]
learner = model_selection.GridSearchCV(svm.SVR(), tuned_parameters, cv=5,
                                       verbose = verbose, n_jobs = -1)
learner.fit(training_set, signal)

标签： pythonscikit-learnsvm

解决方案

推荐阅读