Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/DevSinghSachan/syntax-augmented-bert
/ types & classes
Types & classes
67 in github.com/DevSinghSachan/syntax-augmented-bert
⨍
Functions
302
◇
Types & classes
67
↓ 5 callers
Class
InputExample
A single training/test example for simple sequence classification.
utils/utils.py:24
↓ 4 callers
Class
ScaledEmbedding
Embedding layer that initialises its values to using a truncated normal variable scaled by the inverse of the embedding dimension.
model/graph_encoder.py:54
↓ 2 callers
Class
BertModel
model/syntax_bert.py:313
↓ 2 callers
Class
CRF
model/crf.py:53
↓ 2 callers
Class
DynamicLossScaler
Class that manages dynamic loss scaling. It is recommended to use :class:`DynamicLossScaler` indirectly, by supplying ``dynamic_loss_scale=T
fp16/loss_scaler.py:62
↓ 2 callers
Class
FeaturizedDataLoader
utils/loader.py:281
↓ 2 callers
Class
FeaturizedDataset
utils/loader.py:124
↓ 2 callers
Class
GNNRelationModel
model/graph_encoder.py:110
↓ 2 callers
Class
Ontonotes
This DatasetReader is designed to read in the English OntoNotes v5.0 data in the format used by the CoNLL 2011/2012 shared tasks. In order to
dataset_reader/ontonotes_utils.py:90
↓ 2 callers
Class
SequenceCriteriaCRF
model/loss.py:31
↓ 2 callers
Class
Tree
Reused tree object from stanfordnlp/treelstm.
model/tree.py:11
↓ 1 callers
Class
BertAttention
model/syntax_bert.py:165
↓ 1 callers
Class
BertEncoder
model/syntax_bert.py:206
↓ 1 callers
Class
BertLayer
model/syntax_bert.py:186
↓ 1 callers
Class
BertSelfAttention
model/syntax_bert.py:81
↓ 1 callers
Class
CoNLL2005
This DatasetReader is designed to read in the CoNLL-2005 SRL data. The file path provided to this class can then be any of the train, test or
dataset_reader/conll2005_utils.py:66
↓ 1 callers
Class
CoNLL2005Sentence
A class representing the annotations available for a single CONLL formatted sentence. Parameters ---------- domain_placeholder : ``st
dataset_reader/conll2005_utils.py:13
↓ 1 callers
Class
DenseSparseAdam
NOTE: This class has been copied verbatim from the separate Dense and Sparse versions of Adam in Pytorch. Implements Adam algorithm with
utils/optimizers.py:240
↓ 1 callers
Class
FeedForwardLayer
model/graph_encoder.py:316
↓ 1 callers
Class
GATEncoderLayer
model/graph_encoder.py:271
↓ 1 callers
Class
JointFusionAttention
model/syntax_bert.py:146
↓ 1 callers
Class
LossScaler
Class that manages a static loss scale. This class is intended to interact with :class:`FP16_Optimizer`, and should not be directly manipula
fp16/loss_scaler.py:25
↓ 1 callers
Class
MultipleOptimizer
utils/optimizers.py:202
↓ 1 callers
Class
MultipleScheduler
utils/optimizers.py:185
↓ 1 callers
Class
OntonotesSentence
A class representing the annotations available for a single CONLL formatted sentence. Parameters ---------- document_id : ``str``
dataset_reader/ontonotes_utils.py:18
↓ 1 callers
Class
Optimizer
utils/optimizers.py:31
↓ 1 callers
Class
Pooler
model/syntax_bert.py:267
↓ 1 callers
Class
RelationalBertSelfAttention
model/graph_encoder.py:332
↓ 1 callers
Class
RobertaEmbeddings
Same as BertEmbeddings with a tiny tweak for positional embeddings indexing.
model/syntax_roberta.py:129
↓ 1 callers
Class
RobertaModel
model/syntax_roberta.py:157
↓ 1 callers
Class
tofp16
Utility module that implements:: def forward(self, input): return input.half()
fp16/fp16util.py:22
Class
BertEmbeddings
Construct the embeddings from word, position and token_type embeddings.
model/syntax_roberta.py:89
Class
BertPreTrainedModel
An abstract class to handle weights initialization and a simple interface for dowloading and loading pretrained models.
model/syntax_bert.py:241
Class
CoNLL2005BrownSRLProcessor
utils/utils.py:350
Class
CoNLL2005Reader
This DatasetReader is designed to read in the English OntoNotes v5.0 data for semantic role labelling. It returns a dataset of instances with
dataset_reader/conll2005_reader.py:38
Class
CoNLL2005WSJSRLProcessor
utils/utils.py:313
Class
DataProcessor
Base class for data converters for sequence classification data sets.
utils/utils.py:73
Class
FP16Model
Convert model to half precision in a batchnorm-safe way.
fp16/fp16util.py:86
Class
FP16_Module
fp16/fp16_opt.py:59
Class
FP16_Optimizer
:class:`FP16_Optimizer` is designed to wrap an existing PyTorch optimizer, and manage static or dynamic loss scaling and master weights in a
fp16/fp16_opt.py:74
Class
FreezeWarmupLinearSchedule
Linear warmup and then linear decay. Linearly increases learning rate from 0 to 1 over `warmup_steps` training steps. Linearly decrea
utils/schedulers.py:22
Class
GATEncoder
model/graph_encoder.py:252
Class
GELU
model/graph_encoder.py:72
Class
GNNClassifier
A wrapper classifier for GNNRelationModel.
model/graph_encoder.py:81
Class
GruGateLayer
We make use of the GRUCell in Pytorch. This formulation is somewhat different from the paper: "https://arxiv.org/pdf/1910.06764.pdf" Stab
model/gate.py:37
Class
HighwayGateLayer
model/gate.py:5
Class
InputFeatures
A single set of features of data.
utils/utils.py:50
Class
InputGateLayer
model/gate.py:15
Class
LabelSmoothingLoss
With label smoothing, KL-divergence between q_{smoothed ground truth prob.}(w) and p_{prob. computed by model}(w) is minimized.
model/loss.py:6
Class
MultiplicativeIntegrationGateLayer
model/gate.py:58
Class
NERScore
utils/utils.py:502
Class
NoamAdamSchedule
utils/schedulers.py:5
Class
OntoNotesNERProcessor
Processor for OntoNotes-5.0 NER task
utils/utils.py:141
Class
OntoNotesSRLProcessor
utils/utils.py:257
Class
OntonotesNamedEntityRecognition
This DatasetReader is designed to read in the English OntoNotes v5.0 data for fine-grained named entity recognition. It returns a dataset of
dataset_reader/ontonotes_ner_reader.py:16
Class
RobertaConfig
model/syntax_roberta.py:153
Class
SRLScore
This class uses the external srl-eval.pl script for computing the CoNLL SRL metrics. Note that this metric reads and writes from disk quite a
utils/utils.py:592
Class
SigmoidTanhGateLayer
model/gate.py:25
Class
SrlReader
This DatasetReader is designed to read in the English OntoNotes v5.0 data for semantic role labelling. It returns a dataset of instances with
dataset_reader/ontonotes_srl_reader.py:87
Class
Sst2Processor
Processor for the SST-2 data set (GLUE version).
utils/utils.py:105
Class
SyntaxBertConfig
model/syntax_bert.py:33
Class
SyntaxBertForSequenceClassification
model/syntax_bert.py:451
Class
SyntaxBertForTokenClassification
model/syntax_bert.py:490
Class
SyntaxRobertaConfig
model/syntax_roberta.py:41
Class
SyntaxRobertaForTokenClassification
model/syntax_roberta.py:189
Class
TacredProcessor
Processor for the TACRED relation extraction data set (EMNLP 2017)
utils/utils.py:192
Class
TacredScore
utils/utils.py:416