GridSearchCV учитывает время выполнения модели или он оценивает точность только в соответствии с высокими параметрами? - PullRequest
0 голосов
/ 10 февраля 2020

Я пытаюсь найти лучшую модель для моей проблемы классификации. Я использую GridSearchCV для оптимизации моих моделей (из библиотеки sklearn (DecisionTreeClassifier, RandomForestClassifier, ...). Я хочу знать, учитывает ли это время выполнения модели? Например, я оптимизирую высокие параметры DecisionTreeClassifier, он запускается через несколько секунд с заданной точностью * 1004. * Имеет ли значение время для высоких параметров, которые GridSearch обнаружил и нашел меня?

Надеюсь, это достаточно ясно! Спасибо.

1 Ответ

0 голосов
/ 10 февраля 2020

Это не учитывается, однако вы можете вернуть эту информацию и использовать ее, чтобы найти лучшую модель со словарем, который возвращается cv_results_. Относительно best_estimator_ :

best_estimator_: Оценщик, который был выбран при поиске, т.е. оценщик, который дал наивысшую оценку (или наименьшую потерю, если указана) для пропущенных данных. Недоступно, если refit = False.

...