From 09537fceae52bc9939f4b374f77ea5feacb2bc37 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Fri, 16 Aug 2019 11:23:23 +0200
Subject: [PATCH] Fixed MultiMLP reading from file
---
transition_machine/src/Classifier.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/transition_machine/src/Classifier.cpp b/transition_machine/src/Classifier.cpp
index d56c666..ecceec6 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);
}
--
GitLab