Index

A

additional_metrics (nlpboost.dataset_config.DatasetConfig attribute) additional_params_tokenizer (nlpboost.model_config.ModelConfig attribute) alias (nlpboost.dataset_config.DatasetConfig attribute) aug_kwargs (nlpboost.augmentation.augmenter_config.NLPAugConfig attribute)

augment() (nlpboost.augmentation.TextAugmenterPipeline.NLPAugPipeline method) augment_data (nlpboost.dataset_config.DatasetConfig attribute) augmenter_cls (nlpboost.augmentation.augmenter_config.NLPAugConfig attribute) AutoTrainer (class in nlpboost.autotrainer)

C

callbacks (nlpboost.dataset_config.DatasetConfig attribute) chunks() (in module nlpboost.utils) CkptCleaner (class in nlpboost.ckpt_cleaner) compute_loss() (nlpboost.hftransformers_manager.MultilabelTrainer method) compute_metrics_classification() (in module nlpboost.metrics) compute_metrics_multilabel() (in module nlpboost.metrics) compute_metrics_ner() (in module nlpboost.metrics) compute_metrics_summarization() (in module nlpboost.metrics)

config_num_labels (nlpboost.dataset_config.DatasetConfig attribute) config_problem_type (nlpboost.model_config.ModelConfig attribute) custom_config_class (nlpboost.model_config.ModelConfig attribute) custom_eval_func (nlpboost.dataset_config.DatasetConfig attribute) custom_model_class (nlpboost.model_config.ModelConfig attribute) custom_params_config_model (nlpboost.model_config.ModelConfig attribute) custom_results_getter (nlpboost.model_config.ModelConfig attribute) custom_tokenization_func (nlpboost.model_config.ModelConfig attribute) custom_trainer_cls (nlpboost.model_config.ModelConfig attribute)

D

data_augmentation_steps (nlpboost.dataset_config.DatasetConfig attribute) data_field (nlpboost.dataset_config.DatasetConfig attribute) dataset_name (nlpboost.dataset_config.DatasetConfig attribute) DatasetConfig (class in nlpboost.dataset_config) decoder_name (nlpboost.model_config.ModelConfig attribute)

dict_to_list() (in module nlpboost.utils) direction_optimize (nlpboost.dataset_config.DatasetConfig attribute) do_nothing (nlpboost.model_config.ModelConfig attribute) dropout_field_name (nlpboost.model_config.ModelConfig attribute) dropout_vals (nlpboost.model_config.ModelConfig attribute)

E

early_stopping_summarization (nlpboost.model_config.ModelConfig attribute)

encoder_name (nlpboost.model_config.ModelConfig attribute)

F

files (nlpboost.dataset_config.DatasetConfig attribute) filter_empty() (in module nlpboost.utils)

fix_dir() (nlpboost.ckpt_cleaner.CkptCleaner method) fixed_training_args (nlpboost.dataset_config.DatasetConfig attribute)

G

general_get_test_results() (nlpboost.results_getter.ResultsGetter method) generation_params (nlpboost.model_config.ModelConfig attribute) get_best_name() (nlpboost.ckpt_cleaner.CkptCleaner method) get_dataset_and_tag2id() (nlpboost.hfdatasets_manager.HFDatasetsManager method)

get_model_cls() (nlpboost.hftransformers_manager.HFTransformersManager method) get_tags() (in module nlpboost.utils) get_test_results_qa() (nlpboost.results_getter.ResultsGetter method) get_test_results_summarization() (nlpboost.results_getter.ResultsGetter method) get_windowed_match_context_answer() (in module nlpboost.utils)

H

hf_hub_username (nlpboost.model_config.ModelConfig attribute) hf_load_kwargs (nlpboost.dataset_config.DatasetConfig attribute) HFDatasetsManager (class in nlpboost.hfdatasets_manager)

HFTransformersManager (class in nlpboost.hftransformers_manager) hp_space (nlpboost.model_config.ModelConfig attribute) hp_space_base() (in module nlpboost.default_param_spaces) hp_space_large() (in module nlpboost.default_param_spaces)

I

id_field_qa (nlpboost.dataset_config.DatasetConfig attribute)

is_2sents (nlpboost.dataset_config.DatasetConfig attribute) is_multilabel (nlpboost.dataset_config.DatasetConfig attribute)

J

joinpaths() (in module nlpboost.utils)

L

label_col (nlpboost.dataset_config.DatasetConfig attribute) length_penalty (nlpboost.model_config.ModelConfig attribute) load_config() (nlpboost.hftransformers_manager.HFTransformersManager method) load_data_collator() (nlpboost.hftransformers_manager.HFTransformersManager method)

load_model_init() (nlpboost.hftransformers_manager.HFTransformersManager method) load_tokenizer() (nlpboost.hftransformers_manager.HFTransformersManager method) load_train_args() (nlpboost.hftransformers_manager.HFTransformersManager method) load_trainer() (nlpboost.hftransformers_manager.HFTransformersManager method) loaded_dataset (nlpboost.dataset_config.DatasetConfig attribute)

M

match_questions_multiple_answers() (in module nlpboost.utils) max_length_summary (nlpboost.dataset_config.DatasetConfig attribute) (nlpboost.model_config.ModelConfig attribute) metric_optimize (nlpboost.dataset_config.DatasetConfig attribute) min_length_summary (nlpboost.model_config.ModelConfig attribute) model_cls_summarization (nlpboost.model_config.ModelConfig attribute) ModelConfig (class in nlpboost.model_config) module nlpboost nlpboost.augmentation nlpboost.augmentation.augmenter_config nlpboost.augmentation.TextAugmenterPipeline nlpboost.autotrainer nlpboost.ckpt_cleaner nlpboost.dataset_config nlpboost.default_param_spaces nlpboost.hfdatasets_manager nlpboost.hftransformers_manager nlpboost.metrics nlpboost.metrics_plotter nlpboost.model_config nlpboost.results_getter nlpboost.tokenization_functions nlpboost.utils

multilabel_label_names (nlpboost.dataset_config.DatasetConfig attribute) MultilabelTrainer (class in nlpboost.hftransformers_manager)

N

n_trials (nlpboost.model_config.ModelConfig attribute) name (nlpboost.augmentation.augmenter_config.NLPAugConfig attribute) (nlpboost.model_config.ModelConfig attribute) NLPAugConfig (class in nlpboost.augmentation.augmenter_config) NLPAugPipeline (class in nlpboost.augmentation.TextAugmenterPipeline) nlpboost module nlpboost.augmentation module nlpboost.augmentation.augmenter_config module nlpboost.augmentation.TextAugmenterPipeline module nlpboost.autotrainer module nlpboost.ckpt_cleaner module nlpboost.dataset_config module nlpboost.default_param_spaces module

nlpboost.hfdatasets_manager module nlpboost.hftransformers_manager module nlpboost.metrics module nlpboost.metrics_plotter module nlpboost.model_config module nlpboost.results_getter module nlpboost.tokenization_functions module nlpboost.utils module no_repeat_ngram_size (nlpboost.model_config.ModelConfig attribute) num_beams (nlpboost.model_config.ModelConfig attribute) num_proc (nlpboost.dataset_config.DatasetConfig attribute)

O

only_test (nlpboost.model_config.ModelConfig attribute)

optuna_hp_search() (nlpboost.autotrainer.AutoTrainer method) overwrite_training_args (nlpboost.model_config.ModelConfig attribute)

P

partial_custom_tok_func_call (nlpboost.model_config.ModelConfig attribute) partial_split (nlpboost.dataset_config.DatasetConfig attribute) plot_metrics() (nlpboost.metrics_plotter.ResultsPlotter method) postprocess_qa_predictions() (nlpboost.results_getter.ResultsGetter method)

pre_func (nlpboost.dataset_config.DatasetConfig attribute) prepare_validation_features_squad() (nlpboost.results_getter.ResultsGetter method) pretokenized_dataset (nlpboost.dataset_config.DatasetConfig attribute) proportion (nlpboost.augmentation.augmenter_config.NLPAugConfig attribute) push_to_hub (nlpboost.model_config.ModelConfig attribute)

R

random_init_trials (nlpboost.model_config.ModelConfig attribute) read_metrics() (nlpboost.metrics_plotter.ResultsPlotter method) remove_dirs() (nlpboost.ckpt_cleaner.CkptCleaner method) remove_fields_pre_func (nlpboost.dataset_config.DatasetConfig attribute)

ResultsGetter (class in nlpboost.results_getter) ResultsPlotter (class in nlpboost.metrics_plotter) resume_from_checkpoint (nlpboost.model_config.ModelConfig attribute) retrain_at_end (nlpboost.dataset_config.DatasetConfig attribute)

S

save_best() (nlpboost.ckpt_cleaner.CkptCleaner method) save_dir (nlpboost.model_config.ModelConfig attribute) save_name (nlpboost.model_config.ModelConfig attribute) seed (nlpboost.dataset_config.DatasetConfig attribute) sentence1_field (nlpboost.dataset_config.DatasetConfig attribute)

sentence2_field (nlpboost.dataset_config.DatasetConfig attribute) smoke_test (nlpboost.dataset_config.DatasetConfig attribute) split (nlpboost.dataset_config.DatasetConfig attribute) squad_v2 (nlpboost.dataset_config.DatasetConfig attribute) summary_field (nlpboost.dataset_config.DatasetConfig attribute)

T

task (nlpboost.dataset_config.DatasetConfig attribute) test_batch_size (nlpboost.model_config.ModelConfig attribute) test_size (nlpboost.dataset_config.DatasetConfig attribute) text_field (nlpboost.dataset_config.DatasetConfig attribute) tie_encoder_decoder (nlpboost.model_config.ModelConfig attribute) tokenize_classification() (in module nlpboost.tokenization_functions) tokenize_ner() (in module nlpboost.tokenization_functions)

tokenize_squad() (in module nlpboost.tokenization_functions) tokenize_summarization() (in module nlpboost.tokenization_functions) train_one_model_fixed_params() (nlpboost.autotrainer.AutoTrainer method) train_one_model_optuna() (nlpboost.autotrainer.AutoTrainer method) train_with_fixed_params() (nlpboost.autotrainer.AutoTrainer method) trainer_cls_summarization (nlpboost.model_config.ModelConfig attribute) type_load (nlpboost.dataset_config.DatasetConfig attribute)

V

val_size (nlpboost.dataset_config.DatasetConfig attribute)