#include </home/runner/work/lima/lima/deeplima/libs/tasks/common/train_params.h>

Inheritance diagram for deeplima::train::train_params_t:

Public Member Functions
	train_params_t (std::string device_string="cpu", float learning_rate=0.001, float weight_decay=0.00001, float beta_one=0.9, float beta_two=0.999, size_t max_epochs=100, size_t max_epochs_without_improvement=10, size_t batch_size=4, float input_dropout=0.3)

virtual const std::string &	get_train_set_fn () const

virtual const std::string &	get_dev_set_fn () const

virtual void	guess_data_sets (const std::string &ud_path, const std::string &corpus)

Public Attributes
std::string	m_output_format

std::string	m_output_model_name

std::string	m_input_model_name

std::string	m_train_set_fn

std::string	m_dev_set_fn

std::map< std::string, std::string >	m_tags

std::string	m_device_string

float	m_learning_rate

float	m_weight_decay

float	m_beta_one

float	m_beta_two

float	m_input_dropout_prob

size_t	m_max_epochs

size_t	m_max_epochs_without_improvement

size_t	m_batch_size

std::string	m_optimizers

Detailed Description

Definition at line 19 of file train_params.h.

Constructor & Destructor Documentation

◆ train_params_t()

deeplima::train::train_params_t::train_params_t	(	std::string	device_string = `"cpu"`,
		float	learning_rate = `0.001`,
		float	weight_decay = `0.00001`,
		float	beta_one = `0.9`,
		float	beta_two = `0.999`,
		size_t	max_epochs = `100`,
		size_t	max_epochs_without_improvement = `10`,
		size_t	batch_size = `4`,
		float	input_dropout = `0.3`
	)

inline

Definition at line 43 of file train_params.h.

Member Function Documentation

◆ get_dev_set_fn()

virtual const std::string & deeplima::train::train_params_t::get_dev_set_fn ( ) const

inlinevirtual

Definition at line 69 of file train_params.h.

◆ get_train_set_fn()

virtual const std::string & deeplima::train::train_params_t::get_train_set_fn ( ) const

inlinevirtual

Definition at line 64 of file train_params.h.

◆ guess_data_sets()

void deeplima::train::train_params_t::guess_data_sets	(	const std::string &	ud_path,
		const std::string &	corpus
	)

virtual

Definition at line 17 of file train_params.cpp.

Member Data Documentation

◆ m_batch_size

size_t deeplima::train::train_params_t::m_batch_size

Definition at line 39 of file train_params.h.

◆ m_beta_one

float deeplima::train::train_params_t::m_beta_one

Definition at line 32 of file train_params.h.

◆ m_beta_two

float deeplima::train::train_params_t::m_beta_two

Definition at line 33 of file train_params.h.

◆ m_dev_set_fn

std::string deeplima::train::train_params_t::m_dev_set_fn

Definition at line 25 of file train_params.h.

◆ m_device_string

std::string deeplima::train::train_params_t::m_device_string

Definition at line 28 of file train_params.h.

◆ m_input_dropout_prob

float deeplima::train::train_params_t::m_input_dropout_prob

Definition at line 35 of file train_params.h.

◆ m_input_model_name

std::string deeplima::train::train_params_t::m_input_model_name

Definition at line 23 of file train_params.h.

◆ m_learning_rate

float deeplima::train::train_params_t::m_learning_rate

Definition at line 30 of file train_params.h.

◆ m_max_epochs

size_t deeplima::train::train_params_t::m_max_epochs

Definition at line 37 of file train_params.h.

◆ m_max_epochs_without_improvement

size_t deeplima::train::train_params_t::m_max_epochs_without_improvement

Definition at line 38 of file train_params.h.

◆ m_optimizers

std::string deeplima::train::train_params_t::m_optimizers

Definition at line 41 of file train_params.h.

◆ m_output_format

std::string deeplima::train::train_params_t::m_output_format

Definition at line 21 of file train_params.h.

◆ m_output_model_name

std::string deeplima::train::train_params_t::m_output_model_name

Definition at line 22 of file train_params.h.

◆ m_tags

std::map<std::string, std::string> deeplima::train::train_params_t::m_tags

Definition at line 26 of file train_params.h.

◆ m_train_set_fn

std::string deeplima::train::train_params_t::m_train_set_fn

Definition at line 24 of file train_params.h.

◆ m_weight_decay

float deeplima::train::train_params_t::m_weight_decay

Definition at line 31 of file train_params.h.

The documentation for this struct was generated from the following files:

deeplima/libs/tasks/common/train_params.h
deeplima/libs/tasks/common/train_params.cpp

Public Member Functions

Public Attributes