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

ai pretrain-predict: fix - → _ in cli parsing

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-14 15:12:07 +01:00
parent f97b771922
commit 1876a8883c
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/parse_args.py
View file

@ -25,7 +25,7 @@ For more information, do src/index.py <subcommand> --help.
""")
exit(0)
subcommand = re.sub(r'[^a-z0-9-]', '', sys.argv[1])
subcommand = re.sub(r'-', '_', re.sub(r'[^a-z0-9-]', '', sys.argv[1]))
subcommand_argparser = importlib.import_module(f"subcommands.{subcommand}").parse_args

5
aimodel/src/subcommands/pretrain_predict.py
View file

@ -68,11 +68,16 @@ def run(args):
if filepath_output != "-":
handle = io.open(filepath_output, "w")
i = 0
for rainfall, water in ai.embed(dataset):
handle.write(json.dumps({
"rainfall": rainfall.numpy().tolist(),
"water": water.numpy().tolist()
}, separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422
if i == 0 or i % 1000 == 0:
sys.stderr.write(f"[pretrain:predict] STEP {i}\r")
i += 1
handle.close()