Как сохранить обученную модель дерева решений в ignite? - PullRequest
0 голосов
/ 09 февраля 2019

Ниже приведен фрагмент кода, который я пытаюсь сохранить в моей модели.Но я не могу найти функциональность API saveModel () для хранения модели.

// 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);**

1 Ответ

0 голосов
/ 11 февраля 2019

каждый алгоритм классификации (KNN, ANN, KMeans ...) реализует экспортируемый интерфейс modelFormat, кроме дерева решений, поэтому в этом случае мы можем сохранить его с помощью ModelsComposition (что верно для сценария дерева решений)

Exporterexporter = new FileExporter <> ();

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

...