pmml example

pmml-example.scala

  // load the data and convert it to a Glass object.
  val data: TypedPipe[Glass] = TextLine(args("input"))
    .map{line => Glass.fromTextLine(line)}

  // open the SVM model used to predict the type of glass.
  val svmModel = getClass.getResource(modelFile).openStream()
  val svmModelEvaluator = Predictable.buildEvaluator(svmModel)

  val predictedType: TypedPipe[(Int, GlassType.PredictType)] = data
    // turn the Glass object into a GlassType
    .map(GlassType.fromGlass)

    // predict type
    .map{ meas => (meas.id, meas.predict(svmModelEvaluator)) }

    // write output
    .write(TypedTsv[(Int, GlassType.PredictType)](outputTypeFile))