Class for easily choosing activation functions.
Definition at line 44 of file TActivationChooser.h.
Public Types | |
enum | EActivationType { kLinear = 0 , kSigmoid , kTanh , kReLU , kRadial } |
Public Member Functions | |
TActivationChooser () | |
Default constructor. | |
virtual | ~TActivationChooser () |
Destructor. | |
TActivation * | CreateActivation (const TString &type) const |
instantiate the correct activation object according to the type chosen (given by a TString) | |
TActivation * | CreateActivation (EActivationType type) const |
instantiate the correct activation object according to the type chosen (given as the enumeration type) | |
std::vector< TString > * | GetAllActivationNames () const |
returns the names of all know activation functions | |
Private Member Functions | |
MsgLogger & | Log () const |
message logger | |
Private Attributes | |
TString | fLINEAR |
MsgLogger * | fLogger |
TString | fRADIAL |
TString | fRELU |
TString | fSIGMOID |
TString | fTANH |
#include <TMVA/TActivationChooser.h>
Enumerator | |
---|---|
kLinear | |
kSigmoid | |
kTanh | |
kReLU | |
kRadial |
Definition at line 47 of file TActivationChooser.h.
TMVA::TActivationChooser::TActivationChooser | ( | ) |
Default constructor.
Definition at line 47 of file TActivationChooser.cxx.
|
virtual |
Destructor.
Definition at line 60 of file TActivationChooser.cxx.
TMVA::TActivation * TMVA::TActivationChooser::CreateActivation | ( | const TString & | type | ) | const |
instantiate the correct activation object according to the type chosen (given by a TString)
Definition at line 90 of file TActivationChooser.cxx.
TMVA::TActivation * TMVA::TActivationChooser::CreateActivation | ( | EActivationType | type | ) | const |
instantiate the correct activation object according to the type chosen (given as the enumeration type)
Definition at line 70 of file TActivationChooser.cxx.
std::vector< TString > * TMVA::TActivationChooser::GetAllActivationNames | ( | ) | const |
returns the names of all know activation functions
Definition at line 107 of file TActivationChooser.cxx.
|
inlineprivate |
message logger
Definition at line 70 of file TActivationChooser.h.
|
private |
Definition at line 63 of file TActivationChooser.h.
|
mutableprivate |
Definition at line 69 of file TActivationChooser.h.
|
private |
Definition at line 67 of file TActivationChooser.h.
|
private |
Definition at line 66 of file TActivationChooser.h.
|
private |
Definition at line 64 of file TActivationChooser.h.
|
private |
Definition at line 65 of file TActivationChooser.h.