shapeworks::DeepSSMParameters
Public Functions
Public Attributes
Name | |
---|---|
const std::string | DEEPSSM_SAMPLER_GAUSSIAN_C |
const std::string | DEEPSSM_SAMPLER_MIXTURE_C |
const std::string | DEEPSSM_SAMPLER_KDE_C |
Public Functions Documentation
function DeepSSMParameters
explicit DeepSSMParameters(
ProjectHandle project
)
function save_to_project
void save_to_project()
function get_aug_num_samples
int get_aug_num_samples()
function set_aug_num_samples
void set_aug_num_samples(
int num_samples
)
function get_aug_num_dims
int get_aug_num_dims()
function set_aug_num_dims
void set_aug_num_dims(
int num_dims
)
function get_training_num_dims
int get_training_num_dims()
function set_training_num_dims
void set_training_num_dims(
int num_dims
)
function get_aug_percent_variability
double get_aug_percent_variability()
function set_aug_percent_variability
void set_aug_percent_variability(
double percent_variability
)
function get_aug_sampler_type
std::string get_aug_sampler_type()
function set_aug_sampler_type
void set_aug_sampler_type(
std::string sampler_type
)
function get_training_epochs
int get_training_epochs()
function set_training_epochs
void set_training_epochs(
int epochs
)
function get_training_learning_rate
double get_training_learning_rate()
function set_training_learning_rate
void set_training_learning_rate(
double rate
)
function get_training_decay_learning_rate
bool get_training_decay_learning_rate()
function set_training_decay_learning_rate
void set_training_decay_learning_rate(
bool decay
)
function get_training_fine_tuning
bool get_training_fine_tuning()
function set_training_fine_tuning
void set_training_fine_tuning(
bool fine_tuning
)
function get_training_fine_tuning_epochs
int get_training_fine_tuning_epochs()
function set_training_fine_tuning_epochs
void set_training_fine_tuning_epochs(
int epochs
)
function get_training_fine_tuning_learning_rate
double get_training_fine_tuning_learning_rate()
function set_training_fine_tuning_learning_rate
void set_training_fine_tuning_learning_rate(
double rate
)
function get_training_batch_size
int get_training_batch_size()
function set_training_batch_size
void set_training_batch_size(
int batch_size
)
function get_training_split
double get_training_split()
function set_training_split
void set_training_split(
double value
)
function get_validation_split
double get_validation_split()
function set_validation_split
void set_validation_split(
double value
)
function get_testing_split
double get_testing_split()
function set_testing_split
void set_testing_split(
double value
)
function get_prep_step_complete
bool get_prep_step_complete()
function set_prep_step_complete
void set_prep_step_complete(
bool value
)
function get_prep_stage
int get_prep_stage()
function set_prep_stage
void set_prep_stage(
int stage
)
function get_aug_step_complete
bool get_aug_step_complete()
function set_aug_step_complete
void set_aug_step_complete(
bool value
)
function get_aug_message
std::string get_aug_message()
function set_aug_message
void set_aug_message(
std::string message
)
function get_training_step_complete
bool get_training_step_complete()
function set_training_step_complete
void set_training_step_complete(
bool value
)
function get_training_message
std::string get_training_message()
function set_training_message
void set_training_message(
std::string message
)
function get_spacing
std::vector< double > get_spacing()
function set_spacing
void set_spacing(
std::vector< double > spacing
)
function get_loss_function
std::string get_loss_function()
function set_loss_function
void set_loss_function(
std::string loss_function
)
function get_tl_net_enabled
bool get_tl_net_enabled()
function set_tl_net_enabled
void set_tl_net_enabled(
bool enabled
)
function get_tl_net_ae_epochs
int get_tl_net_ae_epochs()
function set_tl_net_ae_epochs
void set_tl_net_ae_epochs(
int num_epochs
)
function get_tl_net_tf_epochs
int get_tl_net_tf_epochs()
function set_tl_net_tf_epochs
void set_tl_net_tf_epochs(
int num_epochs
)
function get_tl_net_joint_epochs
int get_tl_net_joint_epochs()
function set_tl_net_joint_epochs
void set_tl_net_joint_epochs(
int num_epochs
)
function get_tl_net_alpha
double get_tl_net_alpha()
function set_tl_net_alpha
void set_tl_net_alpha(
double alpha
)
function get_tl_net_a_ae
double get_tl_net_a_ae()
function set_tl_net_a_ae
void set_tl_net_a_ae(
double a_ae
)
function get_tl_net_c_ae
double get_tl_net_c_ae()
function set_tl_net_c_ae
void set_tl_net_c_ae(
double c_ae
)
function get_tl_net_a_lat
double get_tl_net_a_lat()
function set_tl_net_a_lat
void set_tl_net_a_lat(
double a_lat
)
function get_tl_net_c_lat
double get_tl_net_c_lat()
function set_tl_net_c_lat
void set_tl_net_c_lat(
double c_lat
)
function restore_split_defaults
void restore_split_defaults()
function restore_augmentation_defaults
void restore_augmentation_defaults()
function restore_training_defaults
void restore_training_defaults()
function restore_defaults
void restore_defaults()
Public Attributes Documentation
variable DEEPSSM_SAMPLER_GAUSSIAN_C
static const std::string DEEPSSM_SAMPLER_GAUSSIAN_C;
variable DEEPSSM_SAMPLER_MIXTURE_C
static const std::string DEEPSSM_SAMPLER_MIXTURE_C;
variable DEEPSSM_SAMPLER_KDE_C
static const std::string DEEPSSM_SAMPLER_KDE_C;
Updated on 2024-03-17 at 12:58:44 -0600