Skip to content

Instantly share code, notes, and snippets.

View ydennisy's full-sized avatar
😎
Hacking.

Dennis ydennisy

😎
Hacking.
View GitHub Profile
@ydennisy
ydennisy / model.py
Created August 11, 2023 18:44
A siamese network for text embedding.
import tensorflow as tf
from keras_nlp.layers import TransformerDecoder
MAX_LEN, VOCAB_SIZE, EMBED_DIMS = 128, 128, 32
class TokenAndPositionEmbedding(tf.keras.layers.Layer):
def __init__(self, maxlen, vocab_size, embed_dim):
super().__init__()
self.token_emb = tf.keras.layers.Embedding(

Keybase proof

I hereby claim:

  • I am ydennisy on github.
  • I am dennisy (https://keybase.io/dennisy) on keybase.
  • I have a public key ASA4hF41cLFu2zkA_6ciyD4cKZFPsXquwQzu_btkQtIeogo

To claim this, I am signing this object: