2017-07-27 26 views
5

Python (3.6) anaconda (64 bit) casus (3.1.2) kullanıyorum. Bir regresyon problemi için keras (2.0.6) kullanarak bir sinir ağı modeli kurdum (bir cevap, 10 değişken). Ben özellik önem grafik şöyle üretebilir merak ediyorum:Özellik Python'da Keras kullanarak sinir ağındaki Önemi Grafiği

feature importance chart

def base_model(): 
    model = Sequential() 
    model.add(Dense(200, input_dim=10, kernel_initializer='normal', activation='relu')) 
    model.add(Dense(1, kernel_initializer='normal')) 
    model.compile(loss='mean_squared_error', optimizer = 'adam') 
    return model 

clf = KerasRegressor(build_fn=base_model, epochs=100, batch_size=5,verbose=0) 
clf.fit(X_train,Y_train) 

cevap