PredictPrice.scala

import resource._
import ml.combust.bundle.BundleFile
import ml.combust.mleap.runtime.MleapContext.defaultContext
import ml.combust.mleap.runtime.MleapSupport._

import ml.combust.mleap.runtime.serialization.FrameReader


object HelloWorld {
 
  def main(args: Array[String]) {

    /**
      * Loading the model in zip file; deserializing it
      */
    val mleapTransformerLr = (for (bf <- managed(BundleFile("jar:file:/Users/harshvardhan/Downloads/airbnb.model.lr.zip"))) yield {
      bf.loadMleapBundle().get.root
    }).tried.get

    /**
      * Test Data
      */
    val s = scala.io.Source.fromURL("https://s3-us-west-2.amazonaws.com/mleap-demo/frame.json").mkString
    
    val bytes = s.getBytes("UTF-8")

    /**
      * Running the test data against the model to get a prediction
      */
    for (frame <- FrameReader("ml.combust.mleap.json").fromBytes(bytes);
         frameLr <- mleapTransformerLr.transform(frame);
         frameLrSelect <- frameLr.select("price_prediction")) {
      println("Price LR: " + frameLrSelect.dataset(0).getDouble(0))
    }


  }
}