ai pretrain_predict: no need to plot here anymore

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

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 ]))
sys.stderr.write(">>> Complete\n")