%%bash

if [ -e genmodel.jar ]
then
    echo "genmodel.jar exists"
else
    curl http://central.maven.org/maven2/ai/h2o/h2o-genmodel/3.22.0.1/h2o-genmodel-3.22.0.1.jar > genmodel.jar
fi


%classpath add jar genmodel.jar
%%classpath add mvn 
net.sf.opencsv:opencsv:2.3
com.google.code.gson:gson:2.6.2
ai.h2o:deepwater-backend-api:1.0.4


import hex.genmodel.easy.RowData
import hex.genmodel.easy.EasyPredictModelWrapper
import hex.genmodel.MojoModel


var row = RowData()
row.put("sepal.length", "5.0")
row.put("sepal.width", "3.4")
row.put("petal.length", "1.4")
row.put("petal.width", "0.2")

var easyModel = EasyPredictModelWrapper( MojoModel.load("model.mojo"));
var p = easyModel.predictMultinomial(row);

p.label