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

ai: off-by-one

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-02 18:13:48 +01:00
parent 23bcd294c4
commit 88acd54a97
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/subcommands/pretrain.py
View file

@ -53,7 +53,7 @@ def run(args):
feature_dim=args.feature_dim,
shape_rainfall=dataset_metadata["rainfallradar"],
shape_water=[ math.ceil(value * 0.5) for value in dataset_metadata["waterdepth"] ]
shape_water=[ math.ceil(value * 0.5) + 1 for value in dataset_metadata["waterdepth"] ]
)
ai.train(dataset_train, dataset_validate)