diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py index b3c07f2..d5ca3c6 100644 --- a/aimodel/src/lib/dataset/dataset.py +++ b/aimodel/src/lib/dataset/dataset.py @@ -56,4 +56,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8): if __name__ == "__main__": - dataset() \ No newline at end of file + ds_train, ds_validate = dataset() + for thing in ds_validate(): + as_str = str(thing) + print(thing[:100]) \ No newline at end of file