使用scikit-learn

时间:2016-04-13 09:48:35

标签: scikit-learn grid-search

我目前正在Python中实现支持向量回归,我通过GridSearchCV估计参数C和gamma。我最初从大约400种C和γ组合中搜索。这是一个非常详尽的搜索,现在已经在我的计算机上运行了一个多小时。

我希望接收状态更新,例如有多少组合需要测试或类似,因为目前很难判断程序是否正在运行,或者它是刚刚冻结还是其他任何组合。

从我在sci-kit学习文档中读到的内容,我似乎无法找到任何帮助。有没有环绕?

1 个答案:

答案 0 :(得分:5)

GridSearchCVverbose=个关键字。尝试将其设置为例如100。

如果您使用sklearn.cross_validation.cross_val_score评估模型,还可以将其verbose=设置为较高级别。

如果您需要更多细节,还可以“破解”您想要使用的评分对象,使其在每次调用时将分数打印到屏幕或文件,例如:

from sklearn.metrics.scorer import accuracy_scorer

def my_accuracy_scorer(*args):
    score = accuracy_scorer(*args)
    print('score is {}'.format(score))
    return score

通过scoring

将此功能用作cross_val_scoreGridSearchCV中的scoring=my_accuracy_scorer关键字