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:
parent
3611338959
commit
f9d8020f48
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user