persia.embedding

Submodules

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.