|
X # 输入
y # 输出
Xtrain,Xtest,Ytrain,Ytest = TTS(X,y,test_size=0.2,random_state=420)
dnn = DNN(hidden_layer_sizes=(52,),random_state=420).fit(Xtrain,Ytrain)
dnn_pipeline = PMMLPipeline([("dnn",dnn)])
sklearn2pmml(dnn_pipeline,"dnn.pmml")
这样生成的dnn.pmml最终给出的是分类值,我想要概率值,该如何生成pmml文件。
注:本地输出概率值是简单的,dnn_pipeline.predict_proba(X) 即可
|
|