MCPcopy Create free account

hub / github.com/artetxem/undreamt / functions

Functions61 in github.com/artetxem/undreamt

↓ 13 callersMethodsize
(self)
undreamt/data.py:53
↓ 11 callersFunctionadd_optimizer
(module, directions=())
undreamt/train.py:114
↓ 4 callersMethod_train
(self, mode)
undreamt/translator.py:40
↓ 3 callersMethodencode
(self, sentences, train=False)
undreamt/translator.py:48
↓ 3 callersMethodgreedy
(self, sentences, max_ratio=2, train=False)
undreamt/translator.py:75
↓ 3 callersMethodinitial_output
(self, batch_size)
undreamt/decoder.py:51
↓ 3 callersMethodmask
(self, lengths)
undreamt/translator.py:64
↓ 3 callersMethodsentences2ids
(self, sentences, eos=False, sos=False)
undreamt/data.py:40
↓ 3 callersFunctiontokenize
(sentence)
undreamt/data.py:183
↓ 3 callersMethodtotal_time
(self)
undreamt/train.py:358
↓ 2 callersMethod__init__
(self, hidden_size, vocabulary_size, bias=True)
undreamt/generator.py:55
↓ 2 callersMethodbeam_search
(self, sentences, beam_size=12, max_ratio=2, train=False)
undreamt/translator.py:97
↓ 2 callersMethodids2sentences
(self, ids)
undreamt/data.py:50
↓ 2 callersMethodnext_batch
(self, size)
undreamt/data.py:109
↓ 2 callersMethodreset_stats
(self)
undreamt/train.py:347
↓ 2 callersFunctionsave_models
(name)
undreamt/train.py:289
↓ 2 callersMethodscore
(self, src, trg, train=False)
undreamt/translator.py:155
↓ 2 callersMethodwords_per_second
(self)
undreamt/train.py:361
↓ 1 callersMethod__init__
(self, input_size, hidden_size, layers, dropout)
undreamt/decoder.py:57
↓ 1 callersMethod_fill_cache
(self)
undreamt/data.py:77
↓ 1 callersMethod_remove
(self, index)
undreamt/data.py:98
↓ 1 callersMethod_score_length
(self, src, trg, src_min, src_max, trg_min, trg_max)
undreamt/data.py:103
↓ 1 callersMethodids2sentence
(self, ids)
undreamt/data.py:47
↓ 1 callersMethodinitial_hidden
(self, batch_size)
undreamt/encoder.py:58
↓ 1 callersMethodlog
(self, step=0)
undreamt/train.py:409
↓ 1 callersFunctionmain
()
translate.py:21
↓ 1 callersMethodoutput_classes
(self)
undreamt/generator.py:63
↓ 1 callersMethodperplexity
(self)
undreamt/train.py:382
↓ 1 callersMethodperplexity_per_word
(self)
undreamt/train.py:355
↓ 1 callersMethodsentence2ids
(self, sentence, eos=False, sos=False)
undreamt/data.py:31
↓ 1 callersMethodstep
(self)
undreamt/train.py:322
↓ 1 callersMethodtranslate
(self)
undreamt/train.py:389
Method__init__
(self, encoder_embeddings, decoder_embeddings, generator, src_dictionary, trg_dictionary, encoder,
undreamt/translator.py:25
Method__init__
(self, embedding_size, hidden_size, layers=1, dropout=0, input_feeding=True)
undreamt/decoder.py:25
Method__init__
(self, corpus, optimizers, translator, batch_size=50)
undreamt/train.py:315
Method__init__
(self, translator, source, reference, batch_size=50, beam_size=0)
undreamt/train.py:366
Method__init__
(self, name, trainer, validators=(), output_prefix=None, encoding='utf-8')
undreamt/train.py:402
Method__init__
(self, dim, alignment_function='general')
undreamt/attention.py:20
Method__init__
(self, words)
undreamt/data.py:27
Method__init__
(self, src_file, trg_file=None, max_sentence_length=80, cache_size=1000)
undreamt/data.py:66
Method__init__
(self, corpus, translator)
undreamt/data.py:146
Method__init__
(self, embedding_size, hidden_size, bidirectional=False, layers=1, dropout=0)
undreamt/encoder.py:24
Method__init__
(self, hidden_size, embedding_size)
undreamt/generator.py:24
Method__init__
(self, embedding_generator, embeddings)
undreamt/generator.py:42
Functioncpu
(x)
undreamt/devices.py:17
Methodforward
(self, ids, lengths, word_embeddings, hidden, context, context_mask, prev_output, generator)
undreamt/decoder.py:36
Methodforward
(self, input, hidden)
undreamt/decoder.py:66
Methodforward
(self, query, context, mask)
undreamt/attention.py:32
Methodforward
(self, ids, lengths, word_embeddings, hidden)
undreamt/encoder.py:36
Methodforward
(self, hidden, embeddings)
undreamt/generator.py:30
Methodforward
(self, hidden)
undreamt/generator.py:47
Methodforward
(self, hidden)
undreamt/generator.py:60
Functiongpu
(x)
undreamt/devices.py:21
Functionmain_train
()
undreamt/train.py:28
Methodnext_batch
(self, size)
undreamt/data.py:151
Methodoutput_classes
(self)
undreamt/generator.py:37
Methodoutput_classes
(self)
undreamt/generator.py:50
Functionrandom_embeddings
(vocabulary_size, embedding_size)
undreamt/data.py:179
Functionread_embeddings
(file, threshold=0, vocabulary=None)
undreamt/data.py:158
Functionspecial_ids
(ids)
undreamt/data.py:57
Functionword_ids
(ids)
undreamt/data.py:61