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

send correct water shape to

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-02 18:09:36 +01:00
parent ad156a9a00
commit ef0f15960d
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/pretrain.py
View file

@ -1,3 +1,4 @@
import math
import sys
import argparse
from asyncio.log import logger
@ -52,7 +53,7 @@ def run(args):
feature_dim=args.feature_dim,
shape_rainfall=dataset_metadata["rainfallradar"],
shape_water=dataset_metadata["waterdepth"]
shape_water=[ math.floor(value * 0.75) for value in dataset_metadata["waterdepth"] ]
)
ai.train(dataset_train, dataset_validate)