• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• BlockManagerUtils.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.apache.spark.lightgbm.BlockManagerUtils.scala Maven / Gradle / Ivy

// Copyright (C) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in project root for information.

package org.apache.spark.lightgbm

import org.apache.spark.sql.Dataset
import org.apache.spark.storage.BlockManager

object BlockManagerUtils {
  /** Returns the block manager from the dataframe's spark context.
    * @param data The dataframe to get the block manager from.
    * @return The block manager.
    */
  def getBlockManager(data: Dataset[_]): BlockManager = {
    data.sparkSession.sparkContext.env.blockManager
  }
}