package com.intel.daal.examples.neural_networks;
import com.intel.daal.algorithms.neural_networks.*;
import com.intel.daal.algorithms.neural_networks.layers.fullyconnected.*;
import com.intel.daal.algorithms.neural_networks.layers.softmax.*;
import com.intel.daal.algorithms.neural_networks.prediction.PredictionTopology;
import com.intel.daal.algorithms.neural_networks.layers.ForwardLayerDescriptor;
import com.intel.daal.algorithms.neural_networks.layers.NextLayers;
import com.intel.daal.algorithms.neural_networks.layers.ForwardLayer;
import com.intel.daal.examples.utils.Service;
import com.intel.daal.services.DaalContext;
class NeuralNetPredicConfigurator {
public static PredictionTopology configureNet(DaalContext context) {
FullyConnectedForwardBatch fullyconnectedLayer1 = new FullyConnectedForwardBatch(context, Float.class, FullyConnectedMethod.defaultDense, 5);
FullyConnectedForwardBatch fullyconnectedLayer2 = new FullyConnectedForwardBatch(context, Float.class, FullyConnectedMethod.defaultDense, 2);
SoftmaxForwardBatch softmaxLayer = new SoftmaxForwardBatch(context, Float.class, SoftmaxMethod.defaultDense);
PredictionTopology topology = new PredictionTopology(context);
long fc1 = topology.add(fullyconnectedLayer1);
long fc2 = topology.add(fullyconnectedLayer2);
long sm = topology.add(softmaxLayer);
topology.addNext(fc1, fc2);
topology.addNext(fc2, sm);
return topology;
}
}