PySpark：获取Spark ML多层感知器分类器的输出层神经元值

Question

我正在使用Spark ML Multilayer Perceptron Classifier进行二元分类。

struct node *new_node(int key, struct node *left, struct node *right) {
    struct node *this = malloc(sizeof *this);
    this->key = key;
    this->left = left;
    this->right = right;
    return this;
}

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

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

Answer 1

可能我的回答有点晚了。但对于那些新人。

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

这是来自doc

的说明

  

输出层中的节点使用softmax函数