首页 文章

我是新手,试图在scala中生成决策树模型并在java中使用该模型进行预测 . 如何在java中加载该模型?

提问于
浏览
0

我已将scala中生成的模型保存到光盘中 .

val model = DecisionTree.trainClassifier(trainingData,numClasses,categoricalFeaturesInfo,impurity,maxDepth,maxBins)model.save(sc,“C:\ Model”)

1 回答

  • 0

    使用相同的Java导入来加载模型:

    DecisionTreeModel model = DecisionTreeModel.load(sc, "C:\\Model");
    

相关问题