apache-spark - 获取 Spark ML 多层感知器分类器的输出层神经元值

Question

我正在使用 Spark ML 多层感知器分类器进行二元分类。

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

输出层有两个神经元，因为它是一个二元分类问题。现在我想为测试集中的每一行获取两个神经元的值，而不是只获取包含 0 或 1 的预测列。

我在 API 文档中找不到任何内容。

Answer 1

根据documentation , Spark ML 包确实提供 MLP(多层感知器)作为仅分类器(无回归)，因此它将输出层值推送到 softmax 函数以生成二进制输出值:

Nodes in the output layer use softmax function