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

train_predict: fixup

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-20 15:42:33 +01:00
parent cc6679c609
commit aed2348a95
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/subcommands/train_predict.py
View file

@ -91,7 +91,8 @@ def do_png(args, ai, dataset, model_code):
model_params = json.loads(readfile(args.params)) model_params = json.loads(readfile(args.params))
i = 0 i = 0
for rainfall, water in batched_iterator(dataset, tensors_in_item=2, batch_size=model_params["batch_size"]): gen = batched_iterator(dataset, tensors_in_item=2, batch_size=model_params["batch_size"])
for rainfall, water in gen:
water_predict_batch = ai.embed(rainfall) water_predict_batch = ai.embed(rainfall)
for water_predict in water_predict_batch: for water_predict in water_predict_batch:
# [ width, height, softmax_probabilities ] → [ batch, width, height ] # [ width, height, softmax_probabilities ] → [ batch, width, height ]