微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Keras 模型的 YellowbrickTypeError:此估计器不是分类器;尝试使用回归或聚类分数可视化工具

如何解决Keras 模型的 YellowbrickTypeError:此估计器不是分类器;尝试使用回归或聚类分数可视化工具

我有以下 Keras DNN 模型并导入了必要的 Keras 和 Yellowbrick 库:

optimizer = RMSprop(0.001)
finalDNNModel_wrap = KerasClassifier(build_fn=parkOptimalDNN(optimizer),epochs=750,batch_size=10,verbose=0)
finalDNNModel = Sequential()
finalDNNModel.add(Dense(32,input_dim=8,activation='relu'))
finalDNNModel.add(Dense(8,activation='relu'))
finalDNNModel.add(Dense(1,activation='sigmoid'))
#Compile the model
finalDNNModel.compile(loss='binary_crossentropy',optimizer=optimizer,metrics=['accuracy'])
# Fit & Evaluate on the independent validation data set
finalDNNModel.fit(X_pca_train,y_train,batch_size = 10,epochs = 750,verbose = 0)
dnnPrediction = (finalDNNModel.predict(X_pca_validation) > 0.5).astype("int64")
dnnPredictProb = finalDNNModel.predict_proba(X_pca_validation)

我使用 YellowBrick 可视化包进行分类报告,如下所示:

classes = ["Not Parkinson","Parkinson"]
pd.set_option('precision',2)
vizDNN = ClassificationReport(finalDNNModel,classes =  classes,cmap="YlGnBu",is_fitted=True,force_model=True,title="DNN")
vizDNN.score(X_pca_validation,y_validation)
vizDNN.show()

它给出了错误:“YellowbrickTypeError:此估计器不是分类器;请尝试使用回归或聚类分数可视化器!” 有人可以帮忙吗

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。