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

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 保存它（这对于决策树场景是正确的）

Exporter exporter = new FileExporter<>();

((ModelsComposition) mdl).saveModel(exporter, filePath);

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

How to save decision tree trained model in ignite?

java

ignite