Spark MultilayerPerceptronClassifier Scala实现例子及优化算法
发布时间
阅读量:
阅读量
以下是一个使用Scala语言调用Spark MLPC的简单示例,需要特别指出的是,MLPC算法不仅适用于二分类任务,同时也能够应对多分类问题。在配置layers参数时,需确保第一层所包含的神经元数量与数据集的特征维度保持一致,而最后一层的神经元数量则应与数据集中所包含的类别数目相匹配。
package com.spark
import org.apache.spark.ml.classification.MultilayerPerceptronClassifier
import org.apache.spark.ml.evaluation.MulticlassClassificationEvaluator
import org.apache.spark.ml.feature.VectorAssembler
import org.apache.spark.sql.SparkSession
object MultilayerPerceptron {
val spark: SparkSession = SparkSession
.builder()
.master("local")
.appName("MultiLayer Perceptron Classifier")
全部评论 (0)
还没有任何评论哟~
