在Spark上递归构建决策树时,是否需要保存中间数据子集?

我在一个50节点的集群上使用Scala/Spark构建决策树。由于我的数据集相当大(约2TB),我想对其进行并行处理。我的代码如下所示

def buildTree(data: RDD[Array[Double]], numInstances: Int): Node = {    // 基本情况
    if (numInstances < minInstances) {
        return new Node(isLeaf = true)
    }
    /*
     * 为数据中的所有列找到最佳分割
     */
    val leftRDD = data.filter(leftSplitCriteria)
    val rightRDD = data.filter(rightSplitCriteria)
    val subset = Seq(leftRDD, rightRDD)
    val counts = Seq(numLeft, numRight)
    val children = (0 until 2).map(i =>
                    (i,subset(i),counts(i)))
                    .par.map(x => {buildTree(x._2,x._3)})
    return new Node(children(0), children(1), Split)}

我的问题是

  1. Scala作为一种惰性语言,不会立即计算map/filter操作的输出。因此,在构建新节点时,父节点以及父节点的父节点的所有过滤器是否会堆积起来(并递归应用)?
  2. 构建树的最佳并行方法是什么?我是否应该在中间步骤缓存/保存数据集?
  3. 运行此代码时,仅提供num-executers是否足够,或者如果我提供executor-coresdriver-cores等会有所不同?

回答:

我最终通过特征在每个层级并行化了分割查找过程。

参考

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

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