如何在点燃时保存决策树训练的模型？

Question

以下是我要保存模型的一部分代码。但是我无法找到saveModel（）API功能来存储模型。

// Create classification trainer.
DecisionTreeClassificationTrainer trainer = new DecisionTreeClassificationTrainer(10, 0.1);

// Train decision tree model.
Model mdl = trainer.fit(
    ignite,
    dataCache,
    featureExtractor,
    labelExtractor
);
Exporter<DecisionTreeNode, String> exporter = new FileExporter<>();
**((DecisionTreeNode)mdl).saveModel(exporter, filePath);**

Answer 1

除决策树之外，每个分类算法（KNN，ANN，KMeans ...）都实现可导出的modelFormat接口，因此在这种情况下，我们可以使用ModelsComposition保存它（对于决策树方案而言是正确的）

导出器导出器=新的FileExporter <>（）;

（（（ModelsComposition）mdl）.saveModel（exporter，filePath）;