Index

A

avg_loss (hfselect.trainers.Trainer property)

C

collate_fn() (hfselect.dataset.Dataset method) compute_scores() (in module hfselect.esm_logme) compute_task_ranking() (in module hfselect.esm_logme)

convert_legacy_to_new() (hfselect.esm.ESM method) create_config() (hfselect.esm.ESM method) create_embedding_dataset() (in module hfselect.embedding_dataset) custom_message (hfselect.esm.ESMNotInitializedError attribute)

D

Dataset (class in hfselect.dataset) default_message (hfselect.dataset.EmptyDatasetError attribute) (hfselect.esmconfig.InvalidESMConfigError attribute) (hfselect.task_ranking.InvalidTaskRankingError attribute)

E

each_evidence() (in module hfselect.logme) EmbeddingDataset (class in hfselect.embedding_dataset) EmptyDatasetError

ESM (class in hfselect.esm) ESMConfig (class in hfselect.esmconfig) ESMNotInitializedError ESMTrainer (class in hfselect.trainers)

F

fetch_esm_configs() (in module hfselect.utils) fetch_esms() (in module hfselect.utils) find_esm_model_infos() (in module hfselect.utils) find_esm_repo_ids() (in module hfselect.utils)

fit() (hfselect.logme.LogME method) forward() (hfselect.esm.ESM method) from_disk() (hfselect.dataset.Dataset class method) (hfselect.embedding_dataset.EmbeddingDataset class method) from_hugging_face() (hfselect.dataset.Dataset class method)

G

get() (hfselect.esmconfig.ESMConfig method)

get_esm_coverage() (in module hfselect.utils) get_pooled_output() (in module hfselect.model_utils)

H

hfselect module hfselect.dataset module hfselect.embedding_dataset module hfselect.esm module hfselect.esm_logme module hfselect.esmconfig module

hfselect.logme module hfselect.model_utils module hfselect.setup_logger module hfselect.task_ranking module hfselect.trainers module hfselect.utils module

I

InvalidEmbeddingDatasetError InvalidESMConfigError

InvalidTaskRankingError is_initialized (hfselect.esm.ESM property) is_valid (hfselect.esmconfig.ESMConfig property)

L

LogME (class in hfselect.logme)

M

module hfselect hfselect.dataset hfselect.embedding_dataset hfselect.esm hfselect.esm_logme hfselect.esmconfig hfselect.logme hfselect.model_utils hfselect.setup_logger hfselect.task_ranking hfselect.trainers hfselect.utils

N

NoESMsFoundError

P

predict() (hfselect.logme.LogME method)

publish() (hfselect.esm.ESM method)

R

reset() (hfselect.logme.LogME method)

reset_loss() (hfselect.trainers.Trainer method)

S

save() (hfselect.dataset.Dataset method) (hfselect.embedding_dataset.EmbeddingDataset method)

save_pretrained() (hfselect.esm.ESM method)

T

TaskRanking (class in hfselect.task_ranking) to_pandas() (hfselect.task_ranking.TaskRanking method) train_with_embeddings() (hfselect.trainers.ESMTrainer method)

train_with_models() (hfselect.trainers.ESMTrainer method) Trainer (class in hfselect.trainers) truncated_svd() (in module hfselect.logme)