@@ -46,8 +46,8 @@ class SlowPointFinder(base.DSAnalyzer):
4646 - exclude any non-unique fixed points according to a tolerance
4747 - exclude any far-away "outlier" fixed points
4848
49- Parameters
50- ----------
49+ Parameters::
50+
5151 f_cell : callable, function, DynamicalSystem
5252 The target of computing the recurrent units.
5353
@@ -307,8 +307,8 @@ def find_fps_with_gd_method(
307307 ):
308308 """Optimize fixed points with gradient descent methods.
309309
310- Parameters
311- ----------
310+ Parameters::
311+
312312 candidates : ArrayType, dict
313313 The array with the shape of (batch size, state dim) of hidden states
314314 of RNN to start training for fixed points.
@@ -409,8 +409,8 @@ def find_fps_with_opt_solver(
409409 ):
410410 """Optimize fixed points with nonlinear optimization solvers.
411411
412- Parameters
413- ----------
412+ Parameters::
413+
414414 candidates: ArrayType, dict
415415 The candidate (initial) fixed points.
416416 opt_solver: str
@@ -453,8 +453,8 @@ def find_fps_with_opt_solver(
453453 def filter_loss (self , tolerance : float = 1e-5 ):
454454 """Filter fixed points whose speed larger than a given tolerance.
455455
456- Parameters
457- ----------
456+ Parameters::
457+
458458 tolerance: float
459459 Discard fixed points with squared speed larger than this value.
460460 """
@@ -478,8 +478,8 @@ def filter_loss(self, tolerance: float = 1e-5):
478478 def keep_unique (self , tolerance : float = 2.5e-2 ):
479479 """Filter unique fixed points by choosing a representative within tolerance.
480480
481- Parameters
482- ----------
481+ Parameters::
482+
483483 tolerance: float
484484 Tolerance for determination of identical fixed points.
485485 """
@@ -500,8 +500,8 @@ def keep_unique(self, tolerance: float = 2.5e-2):
500500 def exclude_outliers (self , tolerance : float = 1e0 ):
501501 """Exclude points whose closest neighbor is further than threshold.
502502
503- Parameters
504- ----------
503+ Parameters::
504+
505505 tolerance: float
506506 Any point whose closest fixed point is greater than tol is an outlier.
507507 """
@@ -545,8 +545,8 @@ def compute_jacobians(
545545 ):
546546 """Compute the Jacobian matrices at the points.
547547
548- Parameters
549- ----------
548+ Parameters::
549+
550550 points: np.ndarray, bm.ArrayType, jax.ndarray
551551 The fixed points with the shape of (num_point, num_dim).
552552 stack_dict_var: bool
@@ -605,17 +605,17 @@ def compute_jacobians(
605605 def decompose_eigenvalues (matrices , sort_by = 'magnitude' , do_compute_lefts = False ):
606606 """Compute the eigenvalues of the matrices.
607607
608- Parameters
609- ----------
608+ Parameters::
609+
610610 matrices: np.ndarray, bm.ArrayType, jax.ndarray
611611 A 3D array with the shape of (num_matrices, dim, dim).
612612 sort_by: str
613613 The method of sorting.
614614 do_compute_lefts: bool
615615 Compute the left eigenvectors? Requires a pseudo-inverse call.
616616
617- Returns
618- -------
617+ Returns::
618+
619619 decompositions : list
620620 A list of dictionaries with sorted eigenvalues components:
621621 (eigenvalues, right eigenvectors, and left eigenvectors).
0 commit comments