Есть ли способ вернуть гиперпараметры, настроенные Pycaret?

Question

Pycaret автоматически ищет лучшие параметры. Например, приведенные ниже коды назначат 5 автоматически настроенных моделей для tuned_top5.

from pycaret.classification import *
setup(data=train, train_size=.9, target='my_target_feature')
tuned_top5 = [tune_model(model) for model in top5]

Однако мне этого мало. Я хочу знать точные названия и значения гиперпараметров. Например, если этот код настраивает max_depth на 9, я хочу, чтобы напечаталось "max_depth=9" или аналогичный результат.

Есть какой-либо способ сделать это?

Answer 1

Вы можете вытащить имена параметров из моделей и распечатать их напрямую:

      print(tuned_top5.get_all_params())

Если вас интересует только один параметр или краткий список параметров, распечатайте их по отдельности.

      params = tuned_top5.get_all_params()
print("max depth = ', params['max_depth'])

Answer 2

Вы можете использовать

      tuned_best_model.get_params

И он выдаст вам список с гиперпараметрами, которые были выбраны при настройке.

Я надеюсь, что это поможет вам.

Answer 3

Я не знаю, это ваш вопрос, но;

      best=compare_models(exclude=['ransac'])

лучшая переменная получает наиболее успешную модель для регрессии по умолчанию r2, поэтому просто используйте;

      print(best)

так что вы можете видеть, какие параметры используются этой моделью. Вы также можете видеть все гиперметры, когда сохраняете модели, или вы можете распечатать модель, которую вы создали вручную с помощью pycaret. Я надеюсь, что это поможет вам.