mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-04 17:13:02 +00:00
ai dataset: add if main == main
This commit is contained in:
parent
de4c3dab17
commit
e93a95f1b3
1 changed files with 4 additions and 1 deletions
|
@ -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])
|
Loading…
Reference in a new issue