sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-06-30 18:44:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

missing kwargs

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-24 17:51:29 +00:00
parent 1e1d6dd273
commit 105dc5bc56
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/dataset/dataset_mono.py
View file

@ -109,7 +109,7 @@ def dataset_mono(dirpath_input, train_percentage=0.8, **kwargs):
return dataset_train, dataset_validate #, filepaths
def dataset_mono_predict(dirpath_input):
def dataset_mono_predict(dirpath_input, **kwargs):
"""Creates a tf.data.Dataset() for prediction using the contrastive learning model.
Note that this WILL MANGLE THE ORDERING if you set parallel_reads_multiplier to anything other than 0!!