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

ds predict: fix filenames in

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 15:45:22 +01:00
parent e333dcba9c
commit 366db658a8
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -87,9 +87,6 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_m
def dataset_predict(dirpath_input, parallel_reads_multiplier=1.5, prefetch=True):
filepaths = get_filepaths(dirpath_input)
filepaths_count = len(filepaths)
for i in range(len(filepaths)):
filepaths.append(filepaths[-1])
return make_dataset(
filepaths=filepaths,
@ -98,7 +95,7 @@ def dataset_predict(dirpath_input, parallel_reads_multiplier=1.5, prefetch=True)
batch_size=None,
dummy_label=False,
prefetch=prefetch
), filepaths[0:filepaths_count], filepaths_count
)
if __name__ == "__main__":
ds_train, ds_validate = dataset("/mnt/research-data/main/rainfallwater_records-viperfinal/")