如何在Scala上導入保存的ML模型？

Question

我在本地保存了線性回歸模型，希望在以后的應用程序中使用。

如何導入和調用在當前應用程序中未創建的已保存模型？

我在IntelliJ上使用Scala。

這是我保存模型的方式：

LRmodel.write.save(ORCpath+"LinearRegModel")

Answer 1

為了能夠加載模型，您需要事先定義相同的模型。 因此，將模型加載到新環境中並不是直接的。

即。 您不能將包含4個節點的模型加載到包含10個節點的模型中。

您可以在這里看到： https : //spark.apache.org/docs/latest/mllib-linear-methods.html#logistic-regression加載的方法是：

val sameModel = LogisticRegressionModel.load(
   sc,
   "target/tmp/scalaLogisticRegressionWithLBFGSModel"
)

Answer 2

不同的應用程序不是問題。 您可以按照@Wonay的說明簡單地通過保存路徑進行加載。

問題是當您移至另一個文件系統時，例如從本地移至Hadoop ...或僅移至另一台本地PC。 坦率地說，在這種情況下，最好只是在新文件系統上重新生成模型。