如何在 ignite 中保存决策树训练模型？答案

【问题标题】：How to save decision tree trained model in ignite?如何在 ignite 中保存决策树训练模型？
【发布时间】：2019-02-11 16:51:37
【问题描述】：

以下是我试图保存我的模型的一段代码。但我找不到用于存储模型的 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);**

【问题讨论】：

Model 接口从Serializable 接口扩展而来。你可以这样保存和加载吗？

标签： java ignite

【解决方案1】：

每个分类算法（KNN, ANN, KMeans...）除了决策树外都实现了可导出的modelFormat接口，所以在这种情况下我们可以用ModelsComposition保存它（决策树场景是这样）

Exporter 导出器 = new FileExporter();

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

【讨论】：