订阅云计算RSS CSDN首页> 云计算

在Apache Spark上跑Logistic Regression算法

发表于2015-07-24 10:30| 次阅读| 来源TechNobium| 0 条评论| 作者Leonard Giura

摘要:在本文中,你将看到Apache Spark可以用于机器学习的任务,如logistic regression。虽然这只是非分布式的单机环境的Scala shell demo,但是Spark的真正强大在于分布式下的内存并行处理能力。

本文旨在介绍使用机器学习算法,来介绍Apache Spark数据处理引擎。我们一开始会先简单介绍一下Spark,然后我们将开始实践一个机器学习的例子。我们将使用Qualitative Bankruptcy数据集,来自UCI机器学习数据仓库。虽然Spark支持同时Java,Scala,Python和R,在本教程中我们将使用Scala作为编程语言。不用担心你没有使用Scala的经验。练习中的每个代码段,我们都会详细解释一遍。

APACHE SPARK

Apache Spark是一个开源的集群计算框架,用Spark编写的应用程序可以比Hadoop MapReduce范式的速度高100倍以上。Spark的一个主要的特点,基于内存,运行速度快,不仅如此,复杂应用在Spark系统上运行,也比基于磁盘的MapReduce更有效。Spark还旨在更通用,因此它提供了以下库:

  • Spark SQL,处理结构化数据的模块
  • MLlib,可扩展的机器学习库
  • GraphX,图和图的并行计算API
  • Spark Streaming,可扩展的,可容错的流式计算程序

正如已经提到的,Spark支持Java,Scala,Python和R编程语言。它还集成了其他大数据工具。特别是,Spark可以运行在Hadoop集群,可以访问任何数据源,包括Hadoop Cassandra。

Spark核心概念

在一个高的抽象层面,一个Spark的应用程序由一个驱动程序作为入口,在一个集群上运行各种并行操作。驱动程序包含了你的应用程序的main函数,然后将这些应用程序分配给集群成员执行。驱动程序通过SparkContext对象来访问计算集群。对于交互式的shell应用,SparkContext默认可通过sc变量访问。

Spark的一个非常重要的概念是RDD–弹性分布式数据集。这是一个不可改变的对象集合。每个RDD会分成多个分区,每个分区可能在不同的群集节点上参与计算。RDD可以包含任何类型的Java,Scala对象,Python或R,包括用户自定义的类。RDDS的产生有两种基本方式:通过加载外部数据集或分配对象的集合如,list或set。

在创建了RDDs之后,我们可以对RDDs做2种不同类型的操作:

  • Transformations - 转换操作,从一个RDD转换成另外一个RDD
  • Actions - 动作操作,通过RDD计算结果

RDDs通过lazy的方式计算 - 即当RDDs碰到Action操作时,才会开始计算。Spark的Transformations操作,都会积累成一条链,只有当需要数据的时候,才会执行这些Transformations操作。每一次RDD进行Action操作时,RDD都会重新生成。如果你希望某些中间的计算结果能被其他的Action操作复用,那么你需要调用Spark的RDD.persist()来保存中间数据。

Spark支持多种运行模式,你可以使用交互式的Shell,或者单独运行一个standalone的Spark程序。不管哪一种方式,你都会有如下的工作流:

  • 输入数据,用于生成RDD
  • 使用Transformations 操作转换数据集
  • 让Spark保存一些中间计算结果,用于复用计算
  • 使用Action操作,让Spark并行计算。Spark内部会自动优化和运行计算任务。

安装Apache Spark

为了开始使用Spark,需要先从官网下载。选择“Pre-built for Hadoop 2.4 and later”版本然后点击“Direct Download”。如果是Windows用户,建议将Spark放进名字没有空格的文件夹中。比如说,将文件解压到:C:\spark。

正如上面所说的,我们将会使用Scala编程语言。进入Spark的安装路径,运行如下命令:

// Linux and Mac users

bin/spark-shell

// Windows users

bin\spark shell

然后你可以在控制台中看到Scala:

scala>

QUALITATIVE 破产分类

现实生活中的问题是可以用机器学习算法来预测的。我们将试图解决的,通过一个公司的定性信息,预测该公司是否会破产。数据集可以从UCI机器学习库https://archive.ics.uci.edu/ml/datasets/qualitative_bankruptcy下载。在Spark的安装文件夹中,创建一个新的文件夹命名为playground。复制 qualitative_bankruptcy.data.txt文件到这里面。这将是我们的训练数据。

数据集包含250个实例,其中143个实例为非破产,107个破产实例。

每一个实例数据格式如下:

  • 工业风险
  • 管理风险
  • 财务灵活性
  • 信誉
  • 竞争力
  • 经营风险

这些被称为定性参数,因为它们不能被表示为一个数字。每一个参数可以取下以下值:

  • P positive
  • A average
  • N negative

数据集的最后一个列是每个实例的分类:B为破产或NB非破产。

鉴于此数据集,我们必须训练一个模型,它可以用来分类新的数据实例,这是一个典型的分类问题。

解决问题的步骤如下:

  • 从qualitative_bankruptcy.data.txt文件中读取数据
  • 解析每一个qualitative值,并将其转换为double型数值。这是我们的分类算法所需要的
  • 将数据集划分为训练和测试数据集
  • 使用训练数据训练模型
  • 计算测试数据的训练误差

SPARK LOGISTIC REGRESSION

我们将用Spark的逻辑回归算法训练分类模型。如果你想知道更多逻辑回归算法的原理,你可以阅读以下教程http://technobium.com/logistic-regression-using-apache-mahout。

在Spark的Scala Shell中粘贴以下import语句:

import org.apache.spark.mllib.classification.{LogisticRegressionWithLBFGS, LogisticRegressionModel}

import org.apache.spark.mllib.regression.LabeledPoint

import org.apache.spark.mllib.linalg.{Vector, Vectors}

这将导入所需的库。

接下来我们将创建一个Scala函数,将数据集中的qualitative数据转换为Double型数值。键入或粘贴以下代码并回车,在Spark Scala Shell。

def getDoubleValue( input:String ) : Double = {

    var result:Double = 0.0

    if (input == "P")  result = 3.0 

    if (input == "A")  result = 2.0

    if (input == "N")  result = 1.0

    if (input == "NB") result = 1.0

    if (input == "B")  result = 0.0

    return result

   }

如果所有的运行都没有问题,你应该看到这样的输出:

getDoubleValue: (input: String)Double

现在,我们可以读取到qualitative_bankruptcy.data.txt文件中的数据。从Spark的角度来看,这是一个Transformation操作。在这个阶段,数据实际上不被读入内存。如前所述,这是一个lazy的方式执行。实际的读取操作是由count()引发,这是一个Action操作。

val data = sc.textFile("playground/Qualitative_Bankruptcy.data.txt")

data.count()

用我们val关键字声明一个常量data。它是一个包含输入数据所有行的RDD。读操作被SC或sparkcontext上下文变量监听。count操作应返回以下结果:

res0: Long = 250

现在是时候为逻辑回归算法准备数据,将字符串转换为数值型。

val parsedData = data.map{line =

    val parts = line.split(",")

    LabeledPoint(getDoubleValue(parts(6)), Vectors.dense(parts.slice(0,6).map(x =getDoubleValue(x))))

}

在这里,我们声明了另外一个常量,命名为parsedData。对于data变量中的每一行数据,我们将做以下操作:

  • 使用“,”拆分字符串,并获得一个向量,命名为parts
  • 创建并返回一个LabeledPoint对象。每个LabeledPoint包含标签和值的向量。在我们的训练数据,标签或类别(破产或非破产)放在最后一列,数组下标0到6。这是我们使用的parts(6)。在保存标签之前,我们将用getDoubleValue()函数将字符串转换为Double型。其余的值也被转换为Double型数值,并保存在一个名为稠密矢量的数据结构。这也是Spark的逻辑回归算法所需要的数据结构。

Spark支持map()转换操作,Action动作执行时,第一个执行的就是map()。

我们来看看我们准备好的数据,使用take():

parsedData.take(10)

上面的代码,告诉Spark从parsedData数组中取出10个样本,并打印到控制台。一样的,take()操作之前,会先执行map()。输出结果如下:

res5: Array[org.apache.spark.mllib.regression.LabeledPoint] = Array((1.0,[3.0,3.0,2.0,2.0,2.0,3.0]), (1.0,[1.0,1.0,2.0,2.0,2.0,1.0]), (1.0,[2.0,2.0,2.0,2.0,2.0,2.0]), (1.0,[3.0,3.0,3.0,3.0,3.0,3.0]), (1.0,[1.0,1.0,3.0,3.0,3.0,1.0]), (1.0,[2.0,2.0,3.0,3.0,3.0,2.0]), (1.0,[3.0,3.0,2.0,3.0,3.0,3.0]), (1.0,[3.0,3.0,3.0,2.0,2.0,3.0]), (1.0,[3.0,3.0,2.0,3.0,2.0,3.0]), (1.0,[3.0,3.0,2.0,2.0,3.0,3.0]))

接着我们划分一下训练数据和测试数据,将parsedData的60%分为训练数据,40%分为测试数据。

val splits = parsedData.randomSplit(Array(0.6, 0.4), seed = 11L)

val trainingData = splits(0)

val testData = splits(1)

训练数据和测试数据也可以像上面一样,使用take()者count()查看。

激动人心的时刻,我们现在开始使用Spark的LogisticRegressioinWithLBFGS()来训练模型。设置好分类个数,这里是2个(破产和非破产):

val model = new LogisticRegressionWithLBFGS().setNumClasses(2).run(trainingData)

当模型训练完,我们可以使用testData来检验一下模型的出错率。

val labelAndPreds = testData.map { point =

  val prediction = model.predict(point.features)

  (point.label, prediction)

}

val trainErr = labelAndPreds.filter(r = r._1 != r._2).count.toDouble / testData.count

变量labelAndPreds保存了map()转换操作,map()将每一个行转换成二元组。二元组包含了testData的标签数据(point.label,分类数据)和预测出来的分类数据(prediction)。模型使用point.features作为输入数据。

最后一行代码,我们使用filter()转换操作和count()动作操作来计算模型出错率。filter()中,保留预测分类和所属分类不一致的元组。在Scala中_1和_2可以用来访问元组的第一个元素和第二个元素。最后用预测出错的数量除以testData训练集的数量,我们可以得到模型出错率:

trainErr: Double = 0.20430107526881722

总结

在这个教程中,你已经看到了Apache Spark可以用于机器学习的任务,如logistic regression。虽然这只是非分布式的单机环境的Scala shell demo,但是Spark的真正强大在于分布式下的内存并行处理能力。

在大数据领域,Spark是目前最活跃的开源项目,在过去几年已迅速获得关注和发展。在过去的几年里。采访了超过2100受访者,各种各样的使用情况和环境。

[参考资料]

“Learning Spark” by HoldenKarau, Andy Konwinski, Patrick Wendell and Matei Zaharia, O’Reilly Media 2015

Lichman, M. (2013). UCI Machine Learning Repository [http://archive.ics.uci.edu/ml]. Irvine, CA: University of California, School of Information and Computer Science

https://spark.apache.org/docs/latest/mllib-linear-methods.html#logistic-regression

https://spark.apache.org/docs/1.1.0/mllib-data-types.html

https://archive.ics.uci.edu/ml/datasets/Qualitative_Bankruptcy

https://databricks.com/blog/2015/01/27/big-data-projects-are-hungry-for-simpler-and-more-powerful-tools-survey-validates-apache-spark-is-gaining-developer-traction.html

原文来自:LOGISTIC REGRESSION USING APACHE SPARK(译者/施聪羽 审校/朱正贵 责编/仲浩) 

关于译者: 施聪羽,浩渺科技服务端研发工程师,修炼中的码农。


【预告】首届中国人工智能大会(CCAI 2015)将于7月26-27日在北京友谊宾馆召开。机器学习与模式识别、大数据的机遇与挑战、人工智能与认知科学、智能机器人四个主题专家云集。人工智能产品库将同步上线,预约咨询:QQ:1192936057。欢迎关注。

0
0