Advertisement

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)

还没有任何评论哟~