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

pretrain-plot: fix crash; remove water code

Browse source 
the model doesn't save the water encoder at this time
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 16:24:07 +01:00
parent a5f03390ef
commit b31645bd5d
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
aimodel/src/subcommands/pretrain_plot.py
View file

@ -39,19 +39,19 @@ def run(args):
with handle_open(filepath_input, "r") as handle:
for line in handle:
obj = json.loads(line)
embeddings.append(obj["rainfall"])
embeddings.append(obj)
logger.info(">>> Plotting rainfall with UMAP\n")
vis_embeddings(filepath_output_rainfall, np.array(embeddings))
# the model doesn't save the water encoder at this time
# embeddings = []
# with handle_open(filepath_input, "r") as handle:
# for line in handle:
# obj = json.loads(line)
# embeddings.append(obj["water"])
embeddings = []
with handle_open(filepath_input, "r") as handle:
for line in handle:
obj = json.loads(line)
embeddings.append(obj["water"])
logger.info(">>> Plotting water with UMAP\n")
vis_embeddings(filepath_output_water, np.array(embeddings))
# logger.info(">>> Plotting water with UMAP\n")
# vis_embeddings(filepath_output_water, np.array(embeddings))
sys.stderr.write(">>> Complete\n")