ai dataset: add if main == main

This commit is contained in:
Starbeamrainbowlabs 2022-07-26 19:24:40 +01:00
parent de4c3dab17
commit e93a95f1b3
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -56,4 +56,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8):
if __name__ == "__main__":
dataset()
ds_train, ds_validate = dataset()
for thing in ds_validate():
as_str = str(thing)
print(thing[:100])