38 #ifndef ROOT_TMVA_TActivation
41 #ifndef ROOT_TMVA_TActivationIdentity
44 #ifndef ROOT_TMVA_TActivationSigmoid
47 #ifndef ROOT_TMVA_TActivationTanh
50 #ifndef ROOT_TMVA_TActivationReLU
53 #ifndef ROOT_TMVA_TActivationRadial
56 #ifndef ROOT_TMVA_MsgLogger
67 fLogger( new
MsgLogger(
"TActivationChooser") )
91 Log() <<
kFATAL <<
"no Activation function of type " << type <<
" found" <<
Endl;
103 if (type == fLINEAR)
return CreateActivation(kLinear);
104 else if (type == fSIGMOID)
return CreateActivation(kSigmoid);
105 else if (type == fTANH)
return CreateActivation(kTanh);
106 else if (type == fRELU)
return CreateActivation(kReLU);
107 else if (type == fRADIAL)
return CreateActivation(kRadial);
109 Log() <<
kFATAL <<
"no Activation function of type " << type <<
" found" <<
Endl;
114 std::vector<TString>*
119 std::vector<TString>* names =
new std::vector<TString>();
120 names->push_back(fLINEAR);
121 names->push_back(fSIGMOID);
122 names->push_back(fTANH);
123 names->push_back(fRELU);
124 names->push_back(fRADIAL);
MsgLogger & Endl(MsgLogger &ml)
TActivation * CreateActivation(EActivationType type) const
virtual ~TActivationChooser()
std::vector< TString > * GetAllActivationNames() const