def _check_ccpm_args(args):
check_arg(args, 'ccpm_use_shared_embedding', bool)
check_arg(args, 'ccpm_use_project', bool)
check_arg(args, 'ccpm_project_size', int)
check_arg(args, 'ccpm_hidden_units', list)
check_arg(args, 'ccpm_activation_fn', str)
check_arg(args, 'ccpm_dropout', float)
check_arg(args, 'ccpm_batch_norm', bool)
check_arg(args, 'ccpm_layer_norm', bool)
check_arg(args, 'ccpm_use_resnet', bool)
check_arg(args, 'ccpm_use_densenet', bool)
check_arg(args, 'ccpm_kernel_sizes', list)
check_arg(args, 'ccpm_filter_nums', list)
check_arg(args, 'leaky_relu_alpha', float)
check_arg(args, 'swish_beta', float)
def _check_ifm_args(args):
check_arg(args, 'ifm_use_shared_embedding', bool)
check_arg(args, 'ifm_use_project', bool)
check_arg(args, 'ifm_project_size', int)
check_arg(args, 'ifm_hidden_unit', int)
check_arg(args, 'ifm_field_dim', int)
def _check_fwfm_args(args):
check_arg(args, 'fwfm_use_shared_embedding', bool)
check_arg(args, 'fwfm_use_project', bool)
check_arg(args, 'fwfm_project_size', int)
def _check_cin_args(args):
check_arg(args, 'cin_use_shared_embedding', bool)
check_arg(args, 'cin_use_project', bool)
check_arg(args, 'cin_project_size', int)
check_arg(args, 'cin_hidden_feature_maps', list)
check_arg(args, 'cin_split_half', bool)
def _check_fibinet_args(args):
check_arg(args, 'fibinet_use_shared_embedding', bool)
check_arg(args, 'fibinet_use_project', bool)
check_arg(args, 'fibinet_project_size', int)
check_arg(args, 'fibinet_hidden_units', list)
check_arg(args, 'fibinet_activation_fn', str)
check_arg(args, 'fibinet_dropout', float)
check_arg(args, 'fibinet_batch_norm', bool)
check_arg(args, 'fibinet_layer_norm', bool)
check_arg(args, 'fibinet_use_resnet', bool)
check_arg(args, 'fibinet_use_densenet', bool)
check_arg(args, 'fibinet_use_se', bool)
check_arg(args, 'fibinet_use_deep', bool)
check_arg(args, 'fibinet_interaction_type', str)
check_arg(args, 'fibinet_se_interaction_type', str)
check_arg(args, 'fibinet_se_use_shared_embedding', bool)
def _check_cross_args(args):
check_arg(args, 'cross_use_shared_embedding', bool)
check_arg(args, 'cross_use_project', bool)
check_arg(args, 'cross_project_size', int)
check_arg(args, 'cross_num_layers', int)
def _check_ipnn_args(args):
check_arg(args, 'ipnn_use_shared_embedding', bool)
check_arg(args, 'ipnn_use_project', bool)
check_arg(args, 'ipnn_project_size', int)
check_arg(args, 'ipnn_hidden_units', list)
check_arg(args, 'ipnn_activation_fn', str)
check_arg(args, 'ipnn_dropout', float)
check_arg(args, 'ipnn_batch_norm', bool)
check_arg(args, 'ipnn_layer_norm', bool)
check_arg(args, 'ipnn_use_resnet', bool)
check_arg(args, 'ipnn_use_densenet', bool)
check_arg(args, 'ipnn_unordered_inner_product', bool)
check_arg(args, 'ipnn_concat_project', bool)
check_arg(args, 'leaky_relu_alpha', float)
check_arg(args, 'swish_beta', float)
def _check_nifm_args(args):
check_arg(args, 'nifm_use_shared_embedding', bool)
check_arg(args, 'nifm_use_project', bool)
check_arg(args, 'nifm_project_size', int)
check_arg(args, 'nifm_hidden_units', (list, tuple))
check_arg(args, 'nifm_activation_fn', str)
check_arg(args, 'nifm_dropout', float)
check_arg(args, 'nifm_batch_norm', bool)
check_arg(args, 'nifm_layer_norm', bool)
check_arg(args, 'nifm_use_resnet', bool)
check_arg(args, 'nifm_use_densenet', bool)
check_arg(args, 'leaky_relu_alpha', float)
check_arg(args, 'swish_beta', float)
def _check_autoint_args(args):
check_arg(args, 'autoint_use_shared_embedding', bool)
check_arg(args, 'autoint_use_project', bool)
check_arg(args, 'autoint_project_size', int)
check_arg(args, 'autoint_size_per_head', int)
check_arg(args, 'autoint_num_heads', int)
check_arg(args, 'autoint_num_blocks', int)
check_arg(args, 'autoint_dropout', float)
check_arg(args, 'autoint_has_residual', bool)
def _check_fgcnn_args(args):
check_arg(args, 'fgcnn_use_shared_embedding', bool)
check_arg(args, 'fgcnn_use_project', bool)
check_arg(args, 'fgcnn_project_dim', int)
check_arg(args, 'fgcnn_filter_nums', list)
check_arg(args, 'fgcnn_kernel_sizes', list)
check_arg(args, 'fgcnn_pooling_sizes', list)
check_arg(args, 'fgcnn_new_map_sizes', list)