chore: improve notebook for training, less verbose

zama-ai · Jan 12, 2024 · 10419c2 · 10419c2
1 parent f0871d4
commit 10419c2
Show file tree

Hide file tree

Showing 2 changed files with 164 additions and 230 deletions.
diff --git a/docs/advanced_examples/LogisticRegressionTraining.ipynb b/docs/advanced_examples/LogisticRegressionTraining.ipynb
diff --git a/src/concrete/ml/sklearn/linear_model.py b/src/concrete/ml/sklearn/linear_model.py
@@ -508,7 +508,8 @@ def _fit_encrypted(
         X_indexes = numpy.arange(0, len(X))
 
         if self.verbose:
-            print("Training starts")
+            mode_string = " (simulation)" if fhe == "simulate" else ""
+            print(f"Training on encrypted data{mode_string}...")
 
         # Iterate on the training quantized module in the clear
         for iteration_step in range(self.max_iter):
@@ -533,8 +534,6 @@ def _fit_encrypted(
 
             to = time.time()
             # Train the model over one iteration
-            if self.verbose:
-                print("Starting iteration ...")
             weights, bias = self.training_quantized_module.forward(  # type: ignore[assignment]
                 X_batch, y_batch, weights, bias, fhe=fhe
             )