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

ai pretrain_predict: no need to plot here anymore

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-13 19:35:44 +01:00
parent 7685ec3e8b
commit 206257f9f5
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -74,11 +74,6 @@ def run(args):
"water": water.numpy().tolist()
}, separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422
if filepath_output != "-":
handle.close()
sys.stderr.write(">>> Plotting with UMAP\n")
filepath_output_umap = os.path.splitext(filepath_output)[0]+'.png'
vis_embeddings(filepath_output_umap, np.array([ embedding.numpy() for embedding in embeddings ]))
handle.close()
sys.stderr.write(">>> Complete\n")