From a0fa25d875da17149c96aed20ea584c03c47f4dd Mon Sep 17 00:00:00 2001
From: Stefan Wunsch <stefan.wunsch@student.kit.edu>
Date: Wed, 30 Nov 2016 15:48:57 +0100
Subject: [PATCH] TMVA: Fix multiclass classification for two classes in method
DNN
---
tmva/tmva/src/MethodDNN.cxx | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tmva/tmva/src/MethodDNN.cxx b/tmva/tmva/src/MethodDNN.cxx
index 1c49a878eaa..901886df238 100644
--- a/tmva/tmva/src/MethodDNN.cxx
+++ b/tmva/tmva/src/MethodDNN.cxx
@@ -419,9 +419,9 @@ void TMVA::MethodDNN::ProcessOptions()
fLayout = TMVA::MethodDNN::ParseLayoutString (fLayoutString);
size_t inputSize = GetNVariables ();
size_t outputSize = 1;
- if (GetNTargets() != 0) {
+ if (fAnalysisType == Types::kRegression && GetNTargets() != 0) {
outputSize = GetNTargets();
- } else if (DataInfo().GetNClasses() > 2) {
+ } else if (fAnalysisType == Types::kMulticlass && DataInfo().GetNClasses() >= 2) {
outputSize = DataInfo().GetNClasses();
}
--
GitLab