From 5a07220020f6505c6f7784bf656d1bcdd7f03b52 Mon Sep 17 00:00:00 2001 From: Bogdan Teleaga Date: Wed, 2 Dec 2020 18:42:54 +0900 Subject: [PATCH] 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. --- umap/parametric_umap.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/umap/parametric_umap.py b/umap/parametric_umap.py index 082f214a..0e9b5f15 100644 --- 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