3
我正在使用Spark ML多層感知器分類器進行二進制分類。PySpark:獲取Spark ML多層感知器分類器的輸出層神經元值
mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2])
輸出層有兩個神經元,因爲它是一個二元分類問題。現在我想爲測試集中的每個行獲取兩個神經元的值,而不是僅獲取包含0或1的預測列。
我找不到任何可以在API文檔中獲取該值的東西。