Index

_

__call__() (angle_emb.angle.AngleDataCollator method) (angle_emb.angle.AngleLoss method) (angle_emb.angle.Pooler method) (angle_emb.AngleDataCollator method) (angle_emb.AngleLoss method) (angle_emb.CorrelationEvaluator method), [1] (angle_emb.evaluation.CorrelationEvaluator method) (angle_emb.Pooler method)

__cfg (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) __version__ (in module angle_emb), [1] (in module angle_emb.version)

A

A (angle_emb.angle.Prompts attribute) (angle_emb.Prompts attribute) AnglE (class in angle_emb) (class in angle_emb.angle) angle_emb module angle_emb.angle module angle_emb.angle_trainer module angle_emb.base module angle_emb.evaluation module angle_emb.loss module angle_emb.utils module angle_emb.version module angle_loss() (in module angle_emb), [1] (in module angle_emb.loss)

angle_pooling_strategy (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) angle_tau (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) angle_w (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) AngleBase (class in angle_emb), [1] (class in angle_emb.base) AngleDataCollator (class in angle_emb) (class in angle_emb.angle) AngleESETrainer (class in angle_emb) (class in angle_emb.angle) AngleLoss (class in angle_emb) (class in angle_emb.angle) AngleTrainer (class in angle_emb) (class in angle_emb.angle) apply_bfloat16 (in module angle_emb.angle_trainer), [1] apply_ese_pca (angle_emb.angle.AngleESETrainer attribute) (angle_emb.AngleESETrainer attribute) apply_lora (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) args (in module angle_emb.angle_trainer)

B

B (angle_emb.angle.Prompts attribute) (angle_emb.Prompts attribute) batch_size (angle_emb.CorrelationEvaluator attribute), [1] (angle_emb.evaluation.CorrelationEvaluator attribute)

best_corrcoef (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute)

C

C (angle_emb.angle.Prompts attribute) (angle_emb.Prompts attribute) categorical_crossentropy_loss() (in module angle_emb) (in module angle_emb.loss) cfg_file_name (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) cln_w (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) compute_distillation_loss() (angle_emb.angle.AngleTrainer method) (angle_emb.AngleTrainer method) compute_loss() (angle_emb.angle.AngleESETrainer method) (angle_emb.angle.AngleTrainer method) (angle_emb.AngleESETrainer method) (angle_emb.AngleTrainer method) compute_mlm_loss() (angle_emb.angle.AngleTrainer method) (angle_emb.AngleTrainer method)

compute_student_loss() (angle_emb.angle.AngleESETrainer method) (angle_emb.AngleESETrainer method) contrastive_with_negative_loss() (in module angle_emb), [1] (in module angle_emb.loss) CorrelationEvaluator (class in angle_emb), [1] (class in angle_emb.evaluation) cosine_loss() (in module angle_emb), [1] (in module angle_emb.loss) cosine_similarity() (in module angle_emb.utils) cosine_tau (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) cosine_w (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) cuda() (angle_emb.AnglE method) (angle_emb.angle.AnglE method)

D

dataset_format (angle_emb.angle.AngleDataCollator attribute) (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleDataCollator attribute) (angle_emb.AngleLoss attribute)

detect_dataset_format() (in module angle_emb) (in module angle_emb.angle) doc_prompt (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute)

E

encode() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) (angle_emb.AngleBase method), [1] (angle_emb.base.AngleBase method) ese_compression_size (angle_emb.angle.AngleESETrainer attribute) (angle_emb.AngleESETrainer attribute) ese_kl_temperature (angle_emb.angle.AngleESETrainer attribute) (angle_emb.AngleESETrainer attribute)

evaluate() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) evaluate_fn (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute) EvaluateCallback (class in angle_emb) (class in angle_emb.angle)

F

filter_duplicate (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute) find_all_linear_names() (in module angle_emb) (in module angle_emb.utils) find_pth_path() (angle_emb.AnglE static method) (angle_emb.angle.AnglE static method)

fit() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) (angle_emb.AngleBase method), [1] (angle_emb.base.AngleBase method) from_pretrained() (angle_emb.AnglE static method) (angle_emb.angle.AnglE static method)

G

get_pooling() (in module angle_emb) (in module angle_emb.utils)

H

hub_model_id (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute)

hub_private_repo (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute)

I

ibn_tau (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute) ibn_w (angle_emb.angle.AngleLoss attribute) (angle_emb.AngleLoss attribute)

in_batch_negative_loss() (in module angle_emb), [1] (in module angle_emb.loss) is_llm (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute)

K

kbit_post_handle() (angle_emb.AnglE static method) (angle_emb.angle.AnglE static method)

L

labels (angle_emb.CorrelationEvaluator attribute), [1] (angle_emb.evaluation.CorrelationEvaluator attribute) list_all_metrics() (angle_emb.CorrelationEvaluator method), [1] (angle_emb.evaluation.CorrelationEvaluator method) list_prompts() (angle_emb.angle.Prompts class method) (angle_emb.Prompts class method) load_config() (angle_emb.AnglE static method) (angle_emb.angle.AnglE static method)

load_kbit (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) load_mlm_model (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) logger (in module angle_emb) (in module angle_emb.utils) lora_config (in module angle_emb.angle_trainer) loss_fct (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute)

M

main() (in module angle_emb.angle_trainer) max_length (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute) model (angle_emb.angle.EvaluateCallback attribute) (angle_emb.angle.Pooler attribute) (angle_emb.EvaluateCallback attribute) (angle_emb.Pooler attribute) model_kwargs (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute)

module angle_emb angle_emb.angle angle_emb.angle_trainer angle_emb.base angle_emb.evaluation angle_emb.loss angle_emb.utils angle_emb.version

N

n_layers (angle_emb.angle.AngleESETrainer attribute) (angle_emb.AngleESETrainer attribute)

O

on_epoch_end() (angle_emb.angle.EvaluateCallback method) (angle_emb.EvaluateCallback method)

P

pad_token_id (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute) padding (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute) padding_side (angle_emb.angle.Pooler attribute) (angle_emb.Pooler attribute) parser (in module angle_emb.angle_trainer) pca_compress() (angle_emb.angle.AngleESETrainer method) (angle_emb.AngleESETrainer method) pooler (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute)

Pooler (class in angle_emb) (class in angle_emb.angle) pooling_strategy (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) (angle_emb.angle.Pooler attribute) (angle_emb.Pooler attribute) prediction_step() (angle_emb.angle.AngleTrainer method) (angle_emb.AngleTrainer method) Prompts (class in angle_emb) (class in angle_emb.angle) push_to_hub (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute) push_to_hub() (angle_emb.AnglE method) (angle_emb.angle.AnglE method)

Q

query_prompt (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute)

R

return_tensors (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute)

S

sample_from_list() (angle_emb.angle.AngleDataCollator static method) (angle_emb.AngleDataCollator static method) save_config() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) save_dir (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute)

save_pretrained() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) set_device() (in module angle_emb) (in module angle_emb.utils) special_columns (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute)

T

teacher_name_or_path (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute) teacher_pooling_strategy (angle_emb.angle.AngleTrainer attribute) (angle_emb.AngleTrainer attribute) text1 (angle_emb.CorrelationEvaluator attribute), [1] (angle_emb.evaluation.CorrelationEvaluator attribute) text2 (angle_emb.CorrelationEvaluator attribute), [1] (angle_emb.evaluation.CorrelationEvaluator attribute) text_prompt (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute)

to() (angle_emb.AnglE method) (angle_emb.angle.AnglE method) tokenizer (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) (angle_emb.angle.AngleDataCollator attribute) (angle_emb.AngleDataCollator attribute) train_mode (angle_emb.AnglE attribute) (angle_emb.angle.AnglE attribute) truncate_layer() (angle_emb.AnglE method) (angle_emb.angle.AnglE method)

V

valid_ds (angle_emb.angle.EvaluateCallback attribute) (angle_emb.EvaluateCallback attribute)