Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/THUDM/P-tuning
/ types & classes
Types & classes
43 in github.com/THUDM/P-tuning
⨍
Functions
222
◇
Types & classes
43
↓ 10 callers
Class
InputExample
A raw input example consisting of one or two segments of text and a label
PT-Fewshot/pet/utils.py:65
↓ 3 callers
Class
LAMADataset
LAMA/data_utils/dataset.py:15
↓ 2 callers
Class
ContinuousPrompt
PT-Fewshot/pet/wrapper.py:82
↓ 1 callers
Class
DictDataset
A dataset of tensors that uses a dictionary for key-value mappings
PT-Fewshot/pet/utils.py:174
↓ 1 callers
Class
EvalConfig
Configuration for evaluating a model.
PT-Fewshot/pet/config.py:72
↓ 1 callers
Class
InputFeatures
A set of numeric features obtained from an :class:`InputExample`
PT-Fewshot/pet/utils.py:112
↓ 1 callers
Class
LimitedExampleList
PT-Fewshot/data_utils/task_processors.py:47
↓ 1 callers
Class
LogitsList
A list of logits obtained from a finetuned PET model
PT-Fewshot/pet/utils.py:28
↓ 1 callers
Class
PTuneForLAMA
LAMA/p_tuning/modeling.py:15
↓ 1 callers
Class
PromptEncoder
LAMA/p_tuning/prompt_encoder.py:5
↓ 1 callers
Class
TrainConfig
Configuration for training a model.
PT-Fewshot/pet/config.py:26
↓ 1 callers
Class
Trainer
LAMA/cli.py:79
↓ 1 callers
Class
TransformerDecoderWrapper
LAMA/megatron_11b/models.py:64
↓ 1 callers
Class
TransformerModelWrapper
A wrapper around a Transformer-based language model.
PT-Fewshot/pet/wrapper.py:138
↓ 1 callers
Class
WrapperConfig
A configuration for a :class:`TransformerModelWrapper`.
PT-Fewshot/pet/config.py:95
Class
BoolQPVP
PT-Fewshot/data_utils/task_pvps.py:445
Class
BoolQProcessor
Processor for the BoolQ data set.
PT-Fewshot/data_utils/task_processors.py:298
Class
CbPVP
PT-Fewshot/data_utils/task_pvps.py:304
Class
CbProcessor
Processor for the CB data set.
PT-Fewshot/data_utils/task_processors.py:171
Class
CopaPVP
PT-Fewshot/data_utils/task_pvps.py:341
Class
CopaProcessor
Processor for the COPA data set.
PT-Fewshot/data_utils/task_processors.py:337
Class
CopaTaskHelper
A custom task helper for the COPA dataset.
PT-Fewshot/pet/task_helpers.py:96
Class
DataProcessor
Abstract class that provides methods for loading train/dev32/dev/test/unlabeled examples for a given task.
PT-Fewshot/data_utils/task_processors.py:91
Class
MLMPreprocessor
Preprocessor for models pretrained using a masked language modeling objective (e.g., BERT).
PT-Fewshot/pet/preprocessor.py:44
Class
MultiRcPVP
PT-Fewshot/data_utils/task_pvps.py:486
Class
MultiRcProcessor
Processor for the MultiRC data set.
PT-Fewshot/data_utils/task_processors.py:393
Class
MultiRcTaskHelper
A custom task helper for the MultiRC dataset.
PT-Fewshot/pet/task_helpers.py:86
Class
MyTaskDataProcessor
Example for a data processor.
PT-Fewshot/data_utils/custom_task_processor.py:27
Class
MyTaskPVP
Example for a pattern-verbalizer pair (PVP).
PT-Fewshot/data_utils/custom_task_pvp.py:26
Class
PLMInputFeatures
A set of numeric input features for a model pretrained with a permuted language modeling objective.
PT-Fewshot/pet/utils.py:160
Class
PVP
This class contains functions to apply patterns and verbalizers as required by PET. Each task requires its own custom implementation of a PVP
PT-Fewshot/data_utils/task_pvps.py:34
Class
PetConfig
Abstract class for a PET configuration that can be saved to and loaded from a json file.
PT-Fewshot/pet/config.py:6
Class
Preprocessor
A preprocessor that transforms an :class:`InputExample` into a :class:`InputFeatures` object so that it can be processed by the model being u
PT-Fewshot/pet/preprocessor.py:18
Class
RecordProcessor
Processor for the ReCoRD data set.
PT-Fewshot/data_utils/task_processors.py:450
Class
RtePVP
PT-Fewshot/data_utils/task_pvps.py:268
Class
RteProcessor
Processor for the RTE data set.
PT-Fewshot/data_utils/task_processors.py:126
Class
TaskHelper
A helper class that provides custom training and evaluation methods for tasks that do not fit in PETs default schema, for example because the
PT-Fewshot/pet/task_helpers.py:27
Class
TransformerLanguageModelWrapper
LAMA/megatron_11b/models.py:8
Class
WicPVP
PT-Fewshot/data_utils/task_pvps.py:521
Class
WicProcessor
Processor for the WiC data set.
PT-Fewshot/data_utils/task_processors.py:178
Class
WscPVP
PT-Fewshot/data_utils/task_pvps.py:397
Class
WscProcessor
Processor for the WSC data set.
PT-Fewshot/data_utils/task_processors.py:218
Class
WscTaskHelper
A custom task helper for the Wsc dataset.
PT-Fewshot/pet/task_helpers.py:185