python sklearn получить список доступных гиперпараметров для модели - PullRequest
0 голосов
/ 07 февраля 2019

Я использую python со sklearn и хотел бы получить список доступных гиперпараметров для модели. Как это можно сделать?Спасибо

Это должно произойти до того, как я инициализирую модель, когда я пытаюсь использовать

model.get_params() 

Я получаю

TypeError: get_params () отсутствует 1требуемый позиционный аргумент: 'self'

1 Ответ

0 голосов
/ 07 февраля 2019

Это должно сделать это: estimator.get_params(), где estimator - название вашей модели.

Чтобы использовать его на модели, вы можете сделать следующее:

reg = RandomForestRegressor()
params = reg.get_params()
# do something...
reg.set_params(params)
reg.fit(X,  y)

РЕДАКТИРОВАТЬ:

Чтобы получить гиперпараметры модели, прежде чем создавать экземпляр класса:

import inspect
import sklearn

models = [sklearn.ensemble.RandomForestRegressor, sklearn.linear_model.LinearRegression]

for m in models:
    hyperparams = inspect.getargspec(m.__init__).args
    print(hyperparams) # Do something with them here

Гиперпараметры модели передаются в конструктор в sklearn, поэтому мы можем использоватьinspect модель, чтобы увидеть, какие параметры конструктора доступны, и, следовательно, гиперпараметры.Возможно, вам придется отфильтровать некоторые аргументы, которые не являются специфичными для модели, такие как self и n_jobs.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...