diff --git a/transition_machine/src/Classifier.cpp b/transition_machine/src/Classifier.cpp index d56c666257dbe20e1ec81fa9a9ad7d3130dc1797..ecceec664053b783993b285c30618bca1906200e 100644 --- a/transition_machine/src/Classifier.cpp +++ b/transition_machine/src/Classifier.cpp @@ -393,6 +393,9 @@ NeuralNetwork * Classifier::createNeuralNetwork(const std::string & modelFilenam if (topology[0] == 'R') return new ReversedMLP(modelFilename); + if (topology[0] == 'M') + return new MultiMLP(modelFilename); + return new MLP(modelFilename); }