Change default loss_report_frequency for ParametricUMAP

The docs of ParametricUMAP claim a loss_report_frequency of 1, but the parameter is 10 by default. This PR changes the value of the parameter to match the docs.
lmcinnes · Dec 2, 2020 · 5a07220 · 5a07220
1 parent 8b03c48
commit 5a07220
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/parametric_umap.py b/umap/parametric_umap.py
@@ -51,7 +51,7 @@ def __init__(
         parametric_reconstruction=False,
         autoencoder_loss=False,
         reconstruction_validation=None,
-        loss_report_frequency=10,
+        loss_report_frequency=1,
         n_training_epochs=1,
         keras_fit_kwargs={},
         **kwargs