3
Spark ML Multilayer Perceptron Classifierを使用してバイナリ分類を行っています。PySpark:Spark ML Multilayer Perceptron Classifierの出力レイヤーニューロン値を取得する
mlp = MultilayerPerceptronClassifier(labelCol="evt", featuresCol="features", layers=[inputneurons,(inputneurons*2)+1,2])
出力レイヤーは、2進分類問題であるため、2つのニューロンから構成されています。今度は、0または1のいずれかを含む予測列を取得するのではなく、テストセット内の各行に対して2つのニューロンの値を取得します。