我用spark-ml多层感知器分类器进行二值分类。
mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2])输出层有两个神经元，因为它是一个二进制分类问题。现在我想为测试集中的每一行获取两个神经元的值，而不是只获取包含0或1的预测列。
我在api文档中找不到任何东西可以得到它。
我用spark-ml多层感知器分类器进行二值分类。
mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2])输出层有两个神经元，因为它是一个二进制分类问题。现在我想为测试集中的每一行获取两个神经元的值，而不是只获取包含0或1的预测列。
我在api文档中找不到任何东西可以得到它。
1条答案
按热度按时间cunj1qz11#
也许我的回答会晚一点。但对于那些新的。
根据本文档，ml包提供mlp（多层感知器）作为分类器（无回归），因此它将输出层值推送到softmax函数以生成二进制输出值。
这是医生的便条
输出层的节点使用softmax函数