diff --git a/tutorials/CMakeLists.txt b/tutorials/CMakeLists.txt index bdda28f179ec9a64375297fe1ec0939e548fd4f7..07f53f2e356bd96dc7bc7591a3ac910677693f9c 100644 --- a/tutorials/CMakeLists.txt +++ b/tutorials/CMakeLists.txt @@ -569,7 +569,7 @@ file(GLOB long_running RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} ${long_running}) set (multithreaded dataframe/df10[2-7]* multicore/mp103* - tmva/TMVAMultiClass.C + tmva/TMVAMulticlass.C tmva/TMVA_CNN_Classification.C tmva/TMVA_Higgs_Classification.C tmva/TMVA_RNN_Classification.C diff --git a/tutorials/tmva/TMVAMulticlass.C b/tutorials/tmva/TMVAMulticlass.C index 09c70ac2be606729bd526b78a40ff7a0a2c522df..7194af54e357e42605167e85c75e2a5788865472 100644 --- a/tutorials/tmva/TMVAMulticlass.C +++ b/tutorials/tmva/TMVAMulticlass.C @@ -137,7 +137,7 @@ void TMVAMulticlass( TString myMethodList = "" ) if (Use["DL_CPU"]) { TString layoutString("Layout=TANH|100,TANH|50,TANH|10,LINEAR"); TString trainingStrategyString("TrainingStrategy=Optimizer=ADAM,LearningRate=1e-3," - "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100"); + "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100,MaxEpochs=20"); TString nnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=N:" "WeightInitialization=XAVIERUNIFORM:Architecture=GPU"); nnOptions.Append(":"); @@ -149,7 +149,7 @@ void TMVAMulticlass( TString myMethodList = "" ) if (Use["DL_GPU"]) { TString layoutString("Layout=TANH|100,TANH|50,TANH|10,LINEAR"); TString trainingStrategyString("TrainingStrategy=Optimizer=ADAM,LearningRate=1e-3," - "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100"); + "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100,MaxEpochs=20"); TString nnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=N:" "WeightInitialization=XAVIERUNIFORM:Architecture=GPU"); nnOptions.Append(":");