sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 01:12:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

ai dataset: add if main == main

Browse source
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
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
aimodel/src/lib/dataset/dataset.py
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])