Functions — github.com/Helsinki-NLP/prosody

CodeHub MCPcopy Create free account

hub / github.com/Helsinki-NLP/prosody / functions

Functions42 in github.com/Helsinki-NLP/prosody

⨍ Functions 42 ◇ Types & classes 10

↓ 7 callersMethod__init__

(self, device, config, labels=None)

↓ 1 callersMethodcollect_stats

↓ 1 callersMethodconvert_tokens_to_emb_ids

prosody_dataset.py:41

↓ 1 callersMethodget_encoding

↓ 1 callersMethodget_tag

(self, encoding)

↓ 1 callersFunctionload_embeddings

(config, vocab)

prosody_dataset.py:169

↓ 1 callersFunctionmain

↓ 1 callersFunctionrediscretize_tag

(value_prominance, nclasses)

prosody_dataset.py:189

↓ 1 callersMethodsave_stats

↓ 1 callersFunctiontest

(model, iterator, criterion, index_to_tag, device, config)

↓ 1 callersFunctiontest_cont

(model, iterator, criterion, index_to_tag, device, config)

↓ 1 callersFunctiontrain

(model, iterator, optimizer, criterion, device, config)

↓ 1 callersFunctiontrain_cont

(model, iterator, optimizer, criterion, device, config)

↓ 1 callersFunctionvalid

(model, iterator, criterion, index_to_tag, device, config, best_dev_acc, best_dev_epoch, epoch)

↓ 1 callersFunctionvalid_cont

(model, iterator, criterion, index_to_tag, device, config, best_dev_acc, best_dev_epoch, epoch)

Method__getitem__

prosody_dataset.py:45

(self, tagged_sents, tag_to_index, config, word_to_embid=None)

prosody_dataset.py:11

(self, device, config, labels=None)

(self, device, config, vocab_size, labels=None)

(self, module, backward=True)

(self, device, config)

(self, device, config, vocab_size)

(self, device, config, index_to_tag)

(self, device, config, index_to_tag, tag_to_index)

(self, device, config, labels=None)

prosody_dataset.py:38

Methodbackward_hook_fn

(self, module, grad_out, grad_in)

x: (N, T). int64 y: (N, T). int64

x: (N, T). int64 y: (N, T). int64

Methodforward_hook_fn

(self, module, input, output)

Functionload_dataset

prosody_dataset.py:84

Methodload_stats

Functionmake_dirs

prosody_dataset.py:148

Functionweighted_mse_loss