Update documentation of NystroemSketchInfluence regarding block-diago…

…nal and Gauss-Newton approximation
aai-institute · Jun 11, 2024 · 339cb21 · 339cb21
1 parent 9701cc0
commit 339cb21
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/docs/influence/influence_function_model.md b/docs/influence/influence_function_model.md
@@ -206,10 +206,16 @@ if_model = NystroemSketchInfluence(
     model,
     loss,
     rank=10,
-    hessian_regularization=0.0,
+    regularization=0.0,
+    block_structure=BlockMode.FULL,
+    second_order_mode=SecondOrderMode.HESSIAN
 )
 if_model.fit(train_loader)
 ```
+This implementation is capable of using a block-matrix
+approximation, see
+[Block-diagonal approximation](#block-diagonal-approximation), and can handle
+[Gauss-Newton approximation](#gauss-newton-approximation).
 
 ### Inverse Harmonic Mean