Spark机器学习入门实例：大数据集二分类

大数据魔镜最大的数据可视化分析平台

“大数据魔镜”邀请好友注册使用即送10元，上不封顶，千万现金红包等你拿。回复关键词“活动”，即刻领取现金。

本篇教程将引领大家，通过使用spark的机器学习性能和 Scala ，练习一个基于超出内存可加载范围的数据集的逻辑回归分类器（即LR分类器）。

假如你想创建一个机器学习模型，但却发现你的输入数据集与你的计算机内存不相符？对于多机器的计算集群环境中通常可以使用如Hadoop和Apache Spark分布式计算工具。然而，Apache Spark能够在本地机器独立模式上，甚至在当输入数据集大于你的计算机内存时通过创建模型处理你的数据。

在这篇文章里，通过使用一个34.6千兆字节的输入数据集创建一个二进制分类模型，为您展现一个Apache Spark的端对端脚本。

可以在您的计算机上运行进行测试。

1.输入数据和预期结果

本文将使用上文中提及数据集的Posts.xml文件。文件大小是34.6千兆字节，这个xml文件包含stackoverflow.com文章数据作为xml属性：

标题 – 文章标题

主体 – 文章文本

标签 – 文章的标签列表

10+ 更多的xml -我们不需要使用的属性

另外我创建一个较小版本的这种文件，里面只有10个条目或文章。此文件包含一个小尺寸的原始数据集，这个数据是被知识共享许可批准的。

如你所料，这个小文件并不是模型训练的最好的选择（这个小模型训练文件并不是最好的选择），这个文件仅适用于实验数据准备代码。然而，本文中的端对端Spark脚本也适用于这个小文件。

我们的目标是创建一个可基于主体和标题预测文章标签的预测模型。为了精简任务和减少代码数量，我们将联接标题和主体并作为一个单独的文本列。

可想而知，这个模型在stackoverflow.com网站上是怎样工作的——用户键入一个问题，网站自动给予标签建议。

假设我们需要尽可能多的正确的标签，并且用户将消除不必要的标签。由于这个假设我们将选择撤销作为我们的模型高优先级目标。

2.二进制和多标签分类

栈溢出标记预测问题属于多标签分类的一种但并不唯一，因为模型应当预测许多分类。相同的文本将被归类为“Java”和“多线程”。注意多标签分类是不同的问题的一个泛化 ——多分类问题，从一组类预测为仅仅一个类。为了简化我们的第一个Apache Spark问题以及减少代码数量，让我们开始简化问题吧。取代练习一个多标记分类器，我们来对一个给定的标签练习一个简单的二进制分类器。例如，对标签“Java”，创建一个能够预测关于Java语言文章的分类器。

通过使用这个简单方法，可以创建几乎所有常见标签的分类器（Java, C++, Python, multi-threading等）。这是一个简单而易学的方法。然而，在实践中却并不完美，因为拆分预测模型基于各自的分类器，你忽略了类之间的相关性。另一个原因——训练许多分类器可能需要大量的计算。

3.在独立模式下设置和运行Apache Spark

如果你的机器上没有Apache Spark，从Spark官网上可以很容易下载到，请下载使用1.5.1版本。

如果Java在你的计算机上已经安装了，那么你可以准备启动Spark了，如果没有的话——请先安装Java。

在Unix系统和Macs系统上，解压缩该文件并复制到任意目录中。现在这个目录就是Spark的目录了。

运行Spark master：

sbin/start-master.sh

运行spark slave：

sbin/start-slaves.sh

运行Spark shell：

bin/spark-shell

Spark shell可以交互模式运行Scala命令。

如果你工作在Hadoop环境的集群模式下，我假定你已经知道怎样运行Spark shell了。

4.导入库

对于这个端到端场景我们将使用Scala，也是Apache Spark的主要编程语言。

// General purpose libraryimport scala.xml._// Spark data manipulation librariesimport org.apache.spark.sql.catalyst.plans._import org.apache.spark.sql._import org.apache.spark.sql.types._import org.apache.spark.sql.functions._// Spark machine learning librariesimport org.apache.spark.ml.feature.import org.apache.spark.ml.classification .LogisticRegressionimport org.apache.spark.mllib.evaluation .BinaryClassificationMetricsimport org.apache.spark.ml.Pipeline

5.解析XML

我们需要从输入的xml文件中提取主体、文本和标签，并用这些列创建一个单独的data-frame。首先，让我们移除xml的页眉和页脚。我假定输入文件和Spark shell命令位于同一目录下。

Spark具有良好的函数可用于解析json和csv格式。对于Xml，我们需要编写一些额外的代码行，通过规格化编码模式来创建一个数据框架。

注意，Scala语言自动转换所有xml代码，像“”转换到实际标签“”。我们也将连接标题和主体、移除所有的不必要的标签以及来自主体和所有空间副本的新行字符。

val postsRDD = postsXml.map { s =>val xml = XML.loadString(s)val id = (xml \ "@Id").textval tags = (xml \ "@Tags").textval title = (xml \ "@Title").textval body = (xml \ "@Body").textval bodyPlain = ("".r).replaceAllIn(body, " ")val text = (title + " " + bodyPlain).replaceAll("\n", " ").replaceAll("( )+", " ");Row(id, tags, text)}

创建一个data-frame，schema应当应用于RDD（弹性分布式数据集，Resilient Distributed Datasets）

val schemaString = "Id Tags Text"val schema = StructType(schemaString.split(" ").map(fieldName => StructField(fieldName, StringType, true)))val postsDf = sqlContext.createDataFrame(postsRDD, schema)

现在你可以查看一下你的数据框架了。

6.准备练习和测试数据集

下一步——为一个二进制分类器创建一个二进制标签。对这个代码示例，我们将用“java”作为一个标签，我们想要用一个二进制分类器来预测。所有存在“java”标签的行被标记为“1”，否则被标记为“0”。让我们来辨别我们的目标标记“java”和基于这个标记创建二进制标签。

val targetTag = "java"val myudf: (String => Double) = (str: String) => val sqlfunc = udf(myudf)val postsLabeled = postsDf.withColumn("Label", sqlfunc(col("Tags")) )

通过使用新标签，数据集可分为消极的子集和积极的子集。

val positive = postsLabeled.filter('Label > 0.0)val negative = postsLabeled.filter('Label

我们将使用数据的90%用于模型训练和另外10%作为数据集测试。让我们通过独立抽样积极和消极数据集来创建一个训练数据集吧。

val positiveTrain = positive.sample(false, 0.9)val negativeTrain = negative.sample(false, 0.9)val training = positiveTrain.unionAll(negativeTrain)

测试数据集应当包括训练数据集中不包括的所有的行。并且再次地——分别地处理积极样例和消极样例

val negativeTrainTmp = negativeTrain .withColumnRenamed("Label", "Flag").select('Id, 'Flag)val negativeTest = negative.join( negativeTrainTmp, negative("Id") === negativeTrainTmp("Id"), "LeftOuter").filter("Flag is null") .select(negative("Id"), 'Tags, 'Text, 'Label)val positiveTrainTmp = positiveTrain .withColumnRenamed("Label", "Flag") .select('Id, 'Flag)val positiveTest = positive.join( positiveTrainTmp, positive("Id") === positiveTrainTmp("Id"), "LeftOuter").filter("Flag is null") .select(positive("Id"), 'Tags, 'Text, 'Label)val testing = negativeTest.unionAll(positiveTest)

7.训练模型

定义训练参数：

1.特征数目

2.回归参数

3.梯度下降的样本点数目

Spark API 创建一个基于来自 data-frame和训练参数的列的模型 ：

8.测试模型

这是我们最后一段二进制“Java”分类器代码，将返回一个预测值（0.0或1.0）：

val testTitle = "Easiest way to merge a release into one JAR file"val testBoby = """Is there a tool or script which easily merges a bunch of href="http://en.wikipedia.org/wiki/JAR_%28file_format %29" JAR files into one JAR file? A bonus would be to easily set the main-file manifest and make it executable. I would like to run it with something like: As far as I can tell, it has no dependencies which indicates that it shouldn't be an easy single-file tool, but the downloaded ZIP file contains a lot of libraries."""val testText = testTitle + testBodyval testDF = sqlContext .createDataFrame(Seq( (99.0, testText))) .toDF("Label", "Text")val result = model.transform(testDF)val prediction = result.collect()(0)(6) .asInstanceOf[Double]print("Prediction: "+ prediction)

让我们评估基于训练数据集的模型的质量。

val testingResult = model.transform(testing)val testingResultScores = testingResult .select("Prediction", "Label").rdd .map(r => (r(0).asInstanceOf[Double], r(1) .asInstanceOf[Double]))val bc = new BinaryClassificationMetrics(testingResultScores)val roc = bc.areaUnderROCprint("Area under the ROC:" + roc)

如果你使用的是小规模的数据集，那么你的模型质量可能不是最好的。ROC值下的面积将非常低（接近50%），意味着一个差的模型质量。如果使用的是整个的Post.xml数据集，质量就不会这么差，ROC下的面积是0.64。或许你可以通过考虑不同的转换，如TF-IDF和规范化来改善结果，但本文中不再详述。

结语

如果你的数据集超出计算内存承受范围的话，Apache Spark可能是数据处理和机器学习脚本的一个非常好的选择。在一个Hadoop Yarn集群模式环境中使用Spark或许不那么容易，但是，在本地或独立模式下，Spark跟其他分析工具一样简单。

（来源：36大数据）