h20_contextexample.scala

import org.apache.spark.h2o.H2OContext
import org.apache.spark.sql.SparkSession


object PartyStarted extends App {

  val sparkSession = SparkSession.builder
    .master("local[*]")
    .appName("Try H2o")
    .config("spark.sql.crossJoin.enabled", "true")
    .config("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
    .config("spark.driver.memory", "15g")
    .getOrCreate()

  val h2oContext = H2OContext.getOrCreate(sparkSession.sparkContext)

  val dataForPost = sparkSession.read.format("").option("header", value = true).csv("Split.csv")

  val h2oData = h2oContext.asH2OFrame(dataForPost)

}