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: don't pass model_code

Browse source 
it's redundant
This commit is contained in:
Starbeamrainbowlabs 2022-10-26 17:11:36 +01:00
parent 1b489518d0
commit 4f9d543695
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -87,13 +87,13 @@ def run(args):
logger.info(f"Records per file: {args.records_per_file}")
if output_mode == MODE_JSONL:
do_jsonl(args, ai, dataset, args.model_code, model_params)
do_jsonl(args, ai, dataset, model_params)
else:
do_png(args, ai, dataset, args.model_code, model_params)
do_png(args, ai, dataset, model_params)
sys.stderr.write(">>> Complete\n")
def do_png(args, ai, dataset, model_code, model_params):
def do_png(args, ai, dataset, model_params):
if not os.path.exists(os.path.dirname(args.output)):
os.mkdir(os.path.dirname(args.output))
@ -114,7 +114,7 @@ def do_png(args, ai, dataset, model_code, model_params):
segmentation_plot(
water_actual, water_predict,
model_code,
args.model_code,
args.output.replace("+d", str(i))
)