lmcinnes / umap

Uniform Manifold Approximation and Projection
BSD 3-Clause "New" or "Revised" License
7.39k stars 803 forks source link

Is 'n_training_epochs' working for parameteric UMAP? #1084

Open PraljakReps opened 9 months ago

PraljakReps commented 9 months ago

I set n_training_epochs to args.epochs, where args.epochs is 1. Of course, I plan to change this after I complete the fast development. However, it appears that parameteric_umap.fit_transform() leads to 10 epochs regardless of how I setup args.epochs. I shared a couple of pictures.

Here is the initialized boilerplate for parameteric umap training.:

image

Here is the output of the training using args.epochs=1:

image

How do I set a desired number of epochs for training that isn't a multiple of 10?