Types & classes — github.com/atcbosselut/comet-commonsense

CodeHub MCPcopy Create free account

hub / github.com/atcbosselut/comet-commonsense / types & classes

Types & classes28 in github.com/atcbosselut/comet-commonsense

⨍ Functions 222◇ Types & classes 28

↓ 17 callersClassDD

utils/utils.py:183

↓ 10 callersClassTextEncoder

mostly a wrapper for a public python bpe tokenizer

src/data/utils.py:54

↓ 4 callersClassConv1D

src/models/gpt.py:53

↓ 2 callersClassGreedySampler

src/evaluate/sampler.py:26

↓ 2 callersClassLayerNorm

Construct a layernorm module in the OpenAI style \ (epsilon inside the square root).

src/models/gpt.py:36

↓ 2 callersClassOpenAIAdam

Implements Open AI version of Adam algorithm with weight decay fix.

src/train/opt.py:34

↓ 1 callersClassAtomicGenerationEvaluator

src/evaluate/atomic_evaluate.py:12

↓ 1 callersClassAtomicGenerationIteratorTrainer

src/train/atomic_train.py:13

↓ 1 callersClassAttention

src/models/gpt.py:76

↓ 1 callersClassBeamSampler

src/evaluate/sampler.py:181

↓ 1 callersClassBlock

src/models/gpt.py:151

↓ 1 callersClassConceptNetGenerationEvaluator

src/evaluate/conceptnet_evaluate.py:13

↓ 1 callersClassConceptNetGenerationIteratorTrainer

src/train/conceptnet_train.py:16

↓ 1 callersClassConceptNetGenerator

src/evaluate/conceptnet_generate.py:14

↓ 1 callersClassLMHead

Language Model Head for the transformer

src/models/gpt.py:216

↓ 1 callersClassLMModel

Transformer with language model head only

src/models/gpt.py:192

↓ 1 callersClassMLP

src/models/gpt.py:136

↓ 1 callersClassTopKSampler

src/evaluate/sampler.py:97

↓ 1 callersClassTransformerModel

Transformer model

src/models/gpt.py:168

↓ 1 callersClassdotdict

dot.notation access to dictionary attributes

src/models/gpt.py:296

ClassDataLoader

src/data/atomic.py:23

src/evaluate/evaluate.py:8

ClassGenerationDataLoader

src/data/conceptnet.py:70

ClassGenerationDataLoader

src/data/atomic.py:62

src/evaluate/generate.py:55

ClassIteratorTrainer

src/train/train.py:177

src/evaluate/sampler.py:15

src/train/train.py:19