Index

A

ActivationFromFnName (class in quaterion_models.modules.simple) aggregate() (FasttextEncoder class method)

aggregation_options (FasttextEncoder attribute) append() (SequentialHead method)

C

collate_fn() (SimilarityModel class method)

CollateFnType (in module quaterion_models.types)

D

disable_gradients_if_required() (Encoder method) (SwitchEncoder method)

E

embedding_size (Encoder property) (FasttextEncoder property) (SwitchEncoder property) EmptyHead (class in quaterion_models.heads.empty_head) encode() (SimilarityModel method)

Encoder (class in quaterion_models.encoders.encoder) encoder_selection() (SwitchEncoder class method) EncoderHead (class in quaterion_models.heads.encoder_head) extract_meta() (Encoder class method) (SwitchEncoder class method)

F

FasttextEncoder (class in quaterion_models.encoders.extras.fasttext_encoder) forward() (ActivationFromFnName method) (Encoder method) (EncoderHead method) (FasttextEncoder method) (SequentialHead method) (SimilarityModel method) (SwitchEncoder method)

G

GatedHead (class in quaterion_models.heads.gated_head) get_collate_fn() (Encoder method) (FasttextEncoder method) (SimilarityModel method) (SwitchEncoder method) get_config_dict() (EncoderHead method) (SequentialHead method) (SkipConnectionHead method) (SoftmaxEmbeddingsHead method) (StackedProjectionHead method) (WideningHead method)

get_encoders_output_size() (SimilarityModel class method) get_meta_extractor() (Encoder method) get_tokens() (FasttextEncoder class method)

I

inverse_permutation() (in module quaterion_models.encoders.switch_encoder)

L

load() (Encoder class method) (EncoderHead class method) (FasttextEncoder class method) (SequentialHead class method) (SimilarityModel class method) (SwitchEncoder class method)

load_fasttext_model() (in module quaterion_models.encoders.extras.fasttext_encoder)

M

MetricModel (in module quaterion_models.model) module quaterion_models quaterion_models.encoders quaterion_models.encoders.encoder quaterion_models.encoders.extras quaterion_models.encoders.extras.fasttext_encoder quaterion_models.encoders.switch_encoder quaterion_models.heads quaterion_models.heads.empty_head quaterion_models.heads.encoder_head quaterion_models.heads.gated_head quaterion_models.heads.sequential_head quaterion_models.heads.skip_connection_head quaterion_models.heads.softmax_head quaterion_models.heads.stacked_projection_head quaterion_models.heads.widening_head quaterion_models.model quaterion_models.modules quaterion_models.modules.simple quaterion_models.types quaterion_models.utils quaterion_models.utils.classes quaterion_models.utils.tensors

move_to_device() (in module quaterion_models.utils.tensors)

O

output_size (EmptyHead property) (EncoderHead property) (GatedHead property) (SequentialHead property) (SkipConnectionHead property) (SoftmaxEmbeddingsHead property) (StackedProjectionHead property)

Q

quaterion_models module quaterion_models.encoders module quaterion_models.encoders.encoder module quaterion_models.encoders.extras module quaterion_models.encoders.extras.fasttext_encoder module quaterion_models.encoders.switch_encoder module quaterion_models.heads module quaterion_models.heads.empty_head module quaterion_models.heads.encoder_head module quaterion_models.heads.gated_head module quaterion_models.heads.sequential_head module

quaterion_models.heads.skip_connection_head module quaterion_models.heads.softmax_head module quaterion_models.heads.stacked_projection_head module quaterion_models.heads.widening_head module quaterion_models.model module quaterion_models.modules module quaterion_models.modules.simple module quaterion_models.types module quaterion_models.utils module quaterion_models.utils.classes module quaterion_models.utils.tensors module

R

reset_parameters() (GatedHead method) (SkipConnectionHead method)

restore_class() (in module quaterion_models.utils.classes)

S

save() (Encoder method) (EncoderHead method) (FasttextEncoder method) (SequentialHead method) (SimilarityModel method) (SwitchEncoder method) save_class_import() (in module quaterion_models.utils.classes)

SequentialHead (class in quaterion_models.heads.sequential_head) SimilarityModel (class in quaterion_models.model) SkipConnectionHead (class in quaterion_models.heads.skip_connection_head) SoftmaxEmbeddingsHead (class in quaterion_models.heads.softmax_head) StackedProjectionHead (class in quaterion_models.heads.stacked_projection_head) switch_collate_fn() (SwitchEncoder class method) SwitchEncoder (class in quaterion_models.encoders.switch_encoder)

T

TensorInterchange (in module quaterion_models.types) train() (SimilarityModel method) trainable (Encoder property) (FasttextEncoder property) (SwitchEncoder property) training (ActivationFromFnName attribute) (EmptyHead attribute) (Encoder attribute) (EncoderHead attribute) (FasttextEncoder attribute) (GatedHead attribute) (SequentialHead attribute) (SimilarityModel attribute) (SkipConnectionHead attribute) (SoftmaxEmbeddingsHead attribute) (StackedProjectionHead attribute) (SwitchEncoder attribute) (WideningHead attribute)

transform() (EmptyHead method) (EncoderHead method) (GatedHead method) (SequentialHead method) (SkipConnectionHead method) (SoftmaxEmbeddingsHead method) (StackedProjectionHead method)

W

WideningHead (class in quaterion_models.heads.widening_head)