Search in sources :

Example 1 with PMMLEncogNeuralNetworkModel

use of ml.shifu.shifu.core.pmml.PMMLEncogNeuralNetworkModel in project shifu by ShifuML.

the class NNSpecifCreator method build.

@Override
public boolean build(BasicML basicML, Model model) {
    NeuralNetwork nnPmmlModel = (NeuralNetwork) model;
    new PMMLEncogNeuralNetworkModel().adaptMLModelToPMML((BasicNetwork) basicML, nnPmmlModel);
    nnPmmlModel.setOutput(createNormalizedOutput());
    return true;
}
Also used : NeuralNetwork(org.dmg.pmml.neural_network.NeuralNetwork) PMMLEncogNeuralNetworkModel(ml.shifu.shifu.core.pmml.PMMLEncogNeuralNetworkModel)

Example 2 with PMMLEncogNeuralNetworkModel

use of ml.shifu.shifu.core.pmml.PMMLEncogNeuralNetworkModel in project shifu by ShifuML.

the class NNSpecifCreator method build.

@Override
public boolean build(BasicML basicML, Model model, int id) {
    NeuralNetwork nnPmmlModel = (NeuralNetwork) model;
    new PMMLEncogNeuralNetworkModel().adaptMLModelToPMML((BasicNetwork) basicML, nnPmmlModel);
    nnPmmlModel.setOutput(createNormalizedOutput(id));
    return true;
}
Also used : NeuralNetwork(org.dmg.pmml.neural_network.NeuralNetwork) PMMLEncogNeuralNetworkModel(ml.shifu.shifu.core.pmml.PMMLEncogNeuralNetworkModel)

Aggregations

PMMLEncogNeuralNetworkModel (ml.shifu.shifu.core.pmml.PMMLEncogNeuralNetworkModel)2 NeuralNetwork (org.dmg.pmml.neural_network.NeuralNetwork)2