constants.py

import os

ROOTDIR = os.path.dirname(os.path.realpath(__file__))

WANDB_KEY = "..."

VOCAB = {'pad_char': '_', 'start_char': '>', 'end_char': ';', 'max_len': 102, 'vocab_size': 36,
         'pad_idx': 0, 'start_idx': 1, 'end_idx': 35,
         'indices_token': {0: '_', 1: '>', 2: 'C', 3: 'c', 4: '(', 5: ')', 6: 'O', 7: '1', 8: '=', 9: 'N', 10: '2',
                           11: '@', 12: '[', 13: ']', 14: 'H', 15: 'n', 16: '3', 17: 'F', 18: '4', 19: 'S', 20: '/',
                           21: 'Cl', 22: 's', 23: '5', 24: 'o', 25: '#', 26: '\\', 27: 'Br', 28: 'P', 29: '6', 30: 'I',
                           31: '7', 32: '8', 33: 'p', 34: '-', 35: ';'},
         'token_indices': {'_': 0, '>': 1, 'C': 2, 'c': 3, '(': 4, ')': 5, 'O': 6, '1': 7, '=': 8, 'N': 9, '2': 10,
                           '@': 11, '[': 12, ']': 13, 'H': 14, 'n': 15, '3': 16, 'F': 17, '4': 18, 'S': 19, '/': 20,
                           'Cl': 21, 's': 22, '5': 23, 'o': 24, '#': 25, '\\': 26, 'Br': 27, 'P': 28, '6': 29, 'I': 30,
                           '7': 31, '8': 32, 'p': 33, '-': 34, ';': 35}
         }