#include <nnet-trnopts.h>

Collaboration diagram for NnetTrainOptions:

Public Member Functions
	NnetTrainOptions ()

void	Register (OptionsItf *opts)

Public Attributes
BaseFloat	learn_rate

BaseFloat	momentum

BaseFloat	l2_penalty

BaseFloat	l1_penalty

Friends
std::ostream &	operator<< (std::ostream &os, const NnetTrainOptions &opts)

Detailed Description

Definition at line 30 of file nnet-trnopts.h.

Constructor & Destructor Documentation

◆ NnetTrainOptions()

NnetTrainOptions ( )

inline

Definition at line 38 of file nnet-trnopts.h.

                     :
     learn_rate(0.008),
     momentum(0.0),
     l2_penalty(0.0),
     l1_penalty(0.0)
   { }

Member Function Documentation

◆ Register()

void Register ( OptionsItf * opts )

inline

Definition at line 46 of file nnet-trnopts.h.

References OptionsItf::Register().

Referenced by main().

                                   {
     opts->Register("learn-rate", &learn_rate, "Learning rate");
     opts->Register("momentum", &momentum, "Momentum");
     opts->Register("l2-penalty", &l2_penalty, "L2 penalty (weight decay)");
     opts->Register("l1-penalty", &l1_penalty, "L1 penalty (promote sparsity)");
   }

Friends And Related Function Documentation

◆ operator<<

std::ostream& operator<<	(	std::ostream &	os,
		const NnetTrainOptions &	opts
	)

friend

Definition at line 54 of file nnet-trnopts.h.

                                                                               {
     os << "NnetTrainOptions : "
        << "learn_rate" << opts.learn_rate << ", "
        << "momentum" << opts.momentum << ", "
        << "l2_penalty" << opts.l2_penalty << ", "
        << "l1_penalty" << opts.l1_penalty;
     return os;
   }