1
0
mirror of https://github.com/osmarks/nanogpt-experiments.git synced 2024-12-21 23:50:28 +00:00

Fix decode fn in shakespeare_char/prepare.py

This commit is contained in:
venusatuluri 2023-01-21 06:14:16 +00:00
parent 3611338959
commit f9d8020f48

View File

@ -31,7 +31,7 @@ itos = { i:ch for i,ch in enumerate(chars) }
def encode(s): def encode(s):
return [stoi[c] for c in s] # encoder: take a string, output a list of integers return [stoi[c] for c in s] # encoder: take a string, output a list of integers
def decode(l): def decode(l):
''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string return ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
# create the train and test splits # create the train and test splits
n = len(data) n = len(data)