如何在Scikit-learn管道中访问回归器的权重



我使用 Keras 回归器来拟合数据的回归。我使用Scikit-learn包装器和Pipeline首先标准化数据,然后将其安装在Keras回归器上。有点像这样:

from sklearn.grid_search import GridSearchCV
from keras.models import Sequential
from keras.layers import Dense
from keras.wrappers.scikit_learn import KerasRegressor
from sklearn.preprocessing import StandardScaler
from sklearn.pipeline import Pipeline
from sklearn.cross_validation import cross_val_score
from sklearn.cross_validation import KFold
from sklearn.externals import joblib
import cPickle
import pandas as pd
import os
from create_model import *
estimators = []
estimators.append(('standardize', StandardScaler()))
estimators.append(('mlp', KerasRegressor(build_fn=create_model, nb_epoch=50,         batch_size=5, verbose=0, neurons = 1)))
pipeline = Pipeline(estimators)

然后,我通过 GridSearchCv 进行网格搜索以获得最佳拟合,并在变量中获得最佳拟合:

batch_size = [60, 80, 100, 200]
epochs = [2, 4, 6, 8, 10, 50]
neurons = np.arange(3,10,1)
optimizer = ['sgd', 'adam', 'rmsprom']
activation = ['relu', 'tanh']
lr = [0.001, 0.01, 0.1]
param_grid = dict(mlp__neurons = neurons, mlp__batch_size = batch_size, mlp__nb_epoch = epochs, mlp__optimizer = optimizer, mlp__activation = activation, mlp__learn_rate = lr)
grid = GridSearchCV(estimator=pipeline, param_grid=param_grid, cv = kfold,scoring='mean_squared_error')
grid_result = grid.fit(X, Y)
clf = []
clf = grid_result.best_estimator_  

clf 变量有 2 个进程,如管道中定义的那样。我的问题是如何通过get_params函数提取 keras 回归器的权重和偏差以实现最佳拟合 (clf)?

clf.get_params()

我找不到好的文档。

weights = KerasRegressor.model.layers[0].get_weights()[0]biases = KerasRegressor.model.layers[0].get_weights()[1]

相关内容

  • 没有找到相关文章

最新更新