【问题标题】:Does the GridSearchCV take in account the execution time of a model or he only takes his accuracy according to the high-parameters?GridSearchCV 是否考虑了模型的执行时间,或者他只根据高参数考虑他的准确性?
【发布时间】:2020-05-25 19:56:29
【问题描述】:
我正在尝试为我的分类问题找到最佳模型。我正在使用 GridSearchCV 来优化我的模型(来自 sklearn 库(DecisionTreeClassifier、RandomForestClassifier、...)。我想知道它是否考虑了模型的执行时间?
前任。我正在优化 DecisionTreeClassifier 的高参数,他在几秒钟内运行,具有特定的准确性。时间是否与 GridSearch 找到并检索到我的高参数相关?
我希望它足够清楚!
谢谢。
【问题讨论】:
标签:
python
scikit-learn
gridsearchcv
【解决方案1】:
不考虑它,但是您可以返回该信息并使用它来查找带有返回的字典的最佳模型cv_results_。关于best_estimator_:
best_estimator_:
搜索选择的估计器,即在遗漏数据上给出最高分数(或最小损失,如果指定)的估计器。如果 refit=False,则不可用。