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

fixup

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-15 17:09:26 +01:00
parent d5f1a26ba3
commit f036e79098
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/dataset/dataset.py
View file

@ -84,7 +84,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_m
return dataset_train, dataset_validate #, filepaths
def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5, pretrain=False):
def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5, prefetch=False):
filepaths = get_filepaths(dirpath_input)
filepaths_count = len(filepaths)
for i in range(len(filepaths)):
@ -96,7 +96,7 @@ def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5,
batch_size=batch_size,
parallel_reads_multiplier=parallel_reads_multiplier,
dummy_label=False,
pretrain=pretrain
prefetch=prefetch
), filepaths[0:filepaths_count], filepaths_count
if __name__ == "__main__":