我正在使用 Spark ML 多层感知器分类器进行二元分类。
mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2])
输出层有两个神经元，因为它是二元分类问题。现在我想获取测试集中每一行的两个神经元的值，而不是仅仅获取包含 0 或 1 的预测列。
我在 API 文档中找不到任何内容。
根据文档Spark ML 包确实提供 MLP（多层感知器）仅作为分类器（无回归），因此它将输出层值推送到 softmax 函数以生成二进制输出值：
输出层的节点使用softmax函数
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
