File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ def fit_dadvi(
29
29
n_draws : int = 1000 ,
30
30
keep_untransformed : bool = False ,
31
31
optimizer_method : minimize_method = "trust-ncg" ,
32
- use_jacobian : bool = True ,
32
+ use_grad : bool = True ,
33
33
use_hessp : bool = True ,
34
34
** minimize_kwargs ,
35
35
) -> az .InferenceData :
@@ -75,8 +75,9 @@ def fit_dadvi(
75
75
``scipy.optimize.minimize`` function. See the documentation of
76
76
that function for details.
77
77
78
- use_jacobian:
79
- If True, pass the Jacobian function to `scipy.optimize.minimize`.
78
+ use_grad:
79
+ If True, pass the gradient function to
80
+ `scipy.optimize.minimize` (where it is referred to as `jac`).
80
81
81
82
use_hessp:
82
83
If True, pass the hessian vector product to `scipy.optimize.minimize`.
@@ -116,7 +117,7 @@ def fit_dadvi(
116
117
117
118
derivative_kwargs = {}
118
119
119
- if use_jacobian :
120
+ if use_grad :
120
121
derivative_kwargs ["jac" ] = True
121
122
if use_hessp :
122
123
derivative_kwargs ["hessp" ] = f_hessp
You can’t perform that action at this time.
0 commit comments