config.py 424 Bytes
import os

MAX_SEQ_LENGTH = 384
BATCH_SIZE = 64
EPOCHS = 50

BASE_OUTPUT = "output/siamese"

DATASET_PATH = "data/pair_dataset.npz" #path for generated pair dataset
VECTOR_PATH = "data/vectors.npz" #path for feature vectors from code dataset
EMBEDDING_PATH = "data/embedding.npz" #path for embedding vector
MODEL_PATH = os.path.sep.join([BASE_OUTPUT, "siamese_model"])
PLOT_PATH = os.path.sep.join([BASE_OUTPUT, "plot.png"])