SHAP如何解释深度信仰网络

时间:2020-11-12 12:43:01

标签: python dbn

classifier = SupervisedDBNClassification(hidden_layers_structure=[100, 100, 100],
                                     learning_rate_rbm=0.05, 
                                     learning_rate=0.1, 
                                     n_epochs_rbm=10, 
                                     n_iter_backprop=100, 
                                     batch_size=20,  # 32
                                     activation_function='relu',
                                     dropout_p=0.2)
model = classifier.fit(X_train, Y_train)

explainer = shap.DeepExplainer(model, X_train)

https://github.com/albertbup/deep-belief-network

SupervisedDBNClassification来自此链接

0 个答案:

没有答案