Flink HBase输入用于机器学习算法

我想使用Flink-HBase插件来读取数据,这些数据将作为Flink机器学习算法的输入,具体来说是SVM和多元线性回归(MLR)。目前,我先将提取的数据写入临时文件,然后通过libSVM方法读取它,但我认为应该有更高级的方法。

你有代码片段或想法可以分享吗?


回答:

没有必要将数据写入磁盘然后再用MLUtils.readLibSVM读取。原因如下。

MLUtils.readLibSVM期望一个文本文件,每行是一个稀疏特征向量及其关联的标签。它使用以下格式来表示标签-特征向量对:

<line> .=. <label> <feature>:<value> <feature>:<value> ... <feature>:<value> # <info>

其中<feature>是特征向量中后续value的索引。MLUtils.readLibSVM可以读取这种格式的文件,并将每一行转换为LabeledVector实例。因此,读取libSVM文件后,你会得到一个DataSet[LabeledVector],这正是SVMMultipleLinearRegression预测器所需的输入格式。

然而,根据你从HBase获取的数据格式,你首先需要将数据转换为libSVM格式。否则,MLUtils.readLibSVM将无法读取写入的文件。如果你转换了数据,那么你也可以直接将数据转换为DataSet[LabeledVector],并将其用作Flink的ML算法的输入。这避免了不必要的磁盘操作周期。

如果你从HBase获得一个DataSet[String],其中每个字符串都具有libSVM格式(见上面的规格),那么你可以对HBase的DataSet应用以下map操作。

val hbaseInput: DataSet[String] = ...val labelCOODS = hbaseInput.flatMap {  line =>    // 移除所有以'#'开头的注释    val commentFreeLine = line.takeWhile(_ != '#').trim    if(commentFreeLine.nonEmpty) {      val splits = commentFreeLine.split(' ')      val label = splits.head.toDouble      val sparseFeatures = splits.tail      val coos = sparseFeatures.map {        str =>          val pair = str.split(':')          require(            pair.length == 2,             "每个特征条目必须具有<feature>:<value>的形式")          // libSVM索引是从1开始的,但我们期望它是从0开始的          val index = pair(0).toInt - 1          val value = pair(1).toDouble          (index, value)      }      Some((label, coos))    } else {      None    }// 计算向量的最大维度val dimensionDS = labelCOODS.map {  labelCOO =>    labelCOO._2.map( _._1 + 1 ).max}.reduce(scala.math.max(_, _))val labeledVectors: DataSet[LabeledVector] =   labelCOODS.map{ new RichMapFunction[(Double, Array[(Int, Double)]), LabeledVector] {  var dimension = 0  override def open(configuration: Configuration): Unit = {    dimension = getRuntimeContext.getBroadcastVariable(DIMENSION).get(0)  }  override def map(value: (Double, Array[(Int, Double)])): LabeledVector = {    new LabeledVector(value._1, SparseVector.fromCOO(dimension, value._2))  }}}.withBroadcastSet(dimensionDS, DIMENSION)

这将把你的libSVM格式数据转换为LabeledVectors的数据集。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注