persia.embedding

Submodules

• persia.embedding.data
• persia.embedding.optim

Package Contents

class persia.embedding.EmbeddingConfig(emb_initialization=(-0.01, 0.01), admit_probability=1.0, weight_bound=10)

Embedding hyperparameters, argument of EmbeddingCtx.

Parameters:

• emb_initialization (Tuple[float, float], optional) – lower and upper bound of embedding uniform initialization.

• admit_probability (float, optional) – the probability (0<=, <=1) of admitting a new embedding.

• weight_bound (float, optional) – restrict each element value of an embedding in [-weight_bound, weight_bound].

persia.embedding.get_default_embedding_config()

Get default embedding configuration.