edit fitting code

DenisTitovLab · Feb 23, 2024 · a17e346 · a17e346
1 parent e4c2c3a
commit a17e346
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/rate_equation_fitting.jl b/src/rate_equation_fitting.jl
@@ -12,22 +12,21 @@ function fit_rate_equation(
         data::DataFrame,
         metab_names::Tuple,
         param_names::Tuple;
-        n_iter = 20
-        # optimization_kwargs = optimization_kwargs
+        n_iter = 20        # optimization_kwargs = optimization_kwargs
 )
-    train_res = train_rate_equation(
+    train_results = train_rate_equation(
         rate_equation::Function,
         data::DataFrame,
         metab_names::Tuple,
         param_names::Tuple;
         n_iter = n_iter,
-        nt_param_choice = nothing
-        # optimization_kwargs = optimization_kwargs
+        nt_param_choice = nothing        # optimization_kwargs = optimization_kwargs
     )
-    rescaled_params = param_rescaling(train_res[2], param_names)
-    return (loss=train_res[1], params = NamedTuple{param_names}(rescaled_params))
+    rescaled_params = param_rescaling(train_results[2], param_names)
+    return (loss = train_results[1], params = NamedTuple{param_names}(rescaled_params))
 end
 
+
 function train_rate_equation(
         rate_equation::Function,
         data::DataFrame,