refining demo

sandialabs · Sep 9, 2023 · 205bdf0 · 205bdf0
1 parent f5a6946
commit 205bdf0
Show file tree

Hide file tree

Showing 2 changed files with 175 additions and 3,348 deletions.
diff --git a/PyUQTk/PyPCE/pce_tools.py b/PyUQTk/PyPCE/pce_tools.py
@@ -624,7 +624,7 @@ def UQTkOptimizeEta(pc_start, y, x, etas, niter, nfolds, mindex_growth, verbose,
  for eta in etas:
 
  # Obtain coefficients through BCS
- pc_final, c_k = UQTkBCS(pc_start, x_tr, y_tr, eta, niter, mindex_growth, ntry=1)
+ pc_final, c_k, _ = UQTkBCS(pc_start, x_tr, y_tr, eta, niter, mindex_growth, ntry=1, return_sigma2=True)
 
  if verbose > 1:
  print("Fold ", i+1, ", eta ", eta, ", ", len(c_k), " terms retained out of a full basis of size", full_basis_size)