Inference API - Spark 0.4.0 API