Это должно сделать это: 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
.