2

我想用MLeap來部署Spark ML機器學習模型,並用它們來實時預測。Scala to Java 8 MLeap Translation

創作者推出了Scala教程,但我需要支持Java 8代碼庫。

我將如何在Java中實現8以下代碼:

val pipeline = SparkUtil.createPipelineModel(uid = "pipeline", Array(featureModel, rfModel)) 

val sbc = SparkBundleContext() 
for(bf <- managed(BundleFile("jar:file:/tmp/mnist.model.rf.zip"))) { 
     pipeline.writeBundle.save(bf)(sbc).get 
     } 

val bundle = (for(bundleFile <- managed(BundleFile("jar:file:/tmp/simple-spark-pipeline.zip"))) yield { 
    bundleFile.loadMleapBundle().get 
}).opt.get 

回答

0

如果使用普通的星火ML變壓器而已,你可以使用SimpleSparkSerializer輕鬆保存和載入您的模型。

保存:

new SimpleSparkSerializer().serializeToBundle(model, "jar:file:/tmp/model.zip", trainData); 

加載:

Transformer model = new SimpleSparkSerializer().deserializeFromBundle("jar:file:/tmp/model.zip"); 
+0

如果我有自定義估計/變形金剛? – Gevorg