qualia2.text package

Submodules

qualia2.text.cbow module

class qualia2.text.cbow.CBOW(vocab_size, embedding_dim=100)

Bases: qualia2.nn.modules.module.Module

Continuous Bag of Words (CBOW) Model

Args:

vocab_size (int): vocabulary size embedding_dim (int): embedding size

forward(ctx, trg)

qualia2.text.text_util module

qualia2.text.text_util.analogy(a, b, c, word2idx, wordvec, n=5)

Predicts word relationship like a:b = c:? Args:

a (str): input string b (str): input string c (str): input string word2idx (dict): word to index map wordvec (Embedding): ector representation of words n (int): top n similar words to show

qualia2.text.text_util.most_similar(query, word2idx, wordvecs, n=5)

Look up most similar words in the embedding Args:

query (str): query text word2idx (dict): word to index map wordvecs (Embedding): vector representation of words n (int): top n similar words to show

Module contents