3

我正在使用Spark ML多層感知器分類器進行二進制分類。PySpark:獲取Spark ML多層感知器分類器的輸出層神經元值

mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2]) 

輸出層有兩個神經元,因爲它是一個二元分類問題。現在我想爲測試集中的每個行獲取兩個神經元的值,而不是僅獲取包含0或1的預測列。

我找不到任何可以在API文檔中獲取該值的東西。

回答

0

可能我的答案會有點晚。但對於那些新的。

根據this documentation,ML包提供的MLP(多層感知器)僅作爲分類器(無迴歸),因此它將輸出層值推送到softmax函數以生成二進制輸出值。

這是從文檔的註釋

節點在輸出層使用SOFTMAX功能