'add_plot'

Author: smasky

UQPyL/inference/amh.py

@@ -12,7 +12,7 @@ class AMH(InferenceABC):
     # Adaptive Metropolis-Hastings
     # ---------------------------- #
 
-    name = "Adaptive Metropolis-Hastings"
+    name = "AMH"
 
     def __init__(self, nChains: int = 1, warmUp: int = 1000, maxIterTimes: int = 1000, 
                        propDist: Literal['gauss', 'uniform'] = 'gauss',

UQPyL/inference/demc.py

@@ -13,7 +13,7 @@ class DEMC(InferenceABC):
     # Differential Evolution Markov Chain
     # ---------------------------------- #
 
-    name = "Differential Evolution Markov Chain"
+    name = "DEMC"
 
     def __init__(self,  nChains: int = 1, warmUp: int = 1000, 
                         maxIterTimes: int = 1000, 

UQPyL/inference/dream_zs.py

@@ -13,7 +13,7 @@ class DREAM_ZS(InferenceABC):
     # Differential Evolution Adaptive Metropolis Z-score
     # ---------------------------------------- #
 
-    name = "Differential Evolution Adaptive Metropolis Z-score"
+    name = "DREAM-ZS"
 
     def __init__(self, nChains: int = 10, warmUp: int = 1000, 
                        ps: float = 0.1, k: int = 1, jitter: float = 0.1,

UQPyL/inference/mh.py

@@ -13,7 +13,7 @@ class MH(InferenceABC):
     # Metropolis–Hastings
     # ------------------- #
 
-    name = "Metropolis-Hastings"
+    name = "MH"
 
     def __init__(self, nChains: int = 1, warmUp: int = 1000, 
                        propDist: Literal['gauss', 'uniform'] = 'gauss',

UQPyL/inference/mh_gibbs.py

@@ -13,7 +13,7 @@ class MH_Gibbs(InferenceABC):
     # MH-Gibbs
     # --------- #
 
-    name = "MH with Gibbs"
+    name = "MH-Gibbs"
 
     def __init__(self, nChains: int = 1, warmUp: int = 1000, maxIters: int = 1000, 
                        propDist: Literal['gauss', 'uniform'] = 'gauss',

UQPyL/util/__init__.py

@@ -3,7 +3,8 @@
 from .poly import PolyFeature
 from .metric import r_square, rank_score, nse, mse, sort_score
 from .verbose import Verbose
-from .plot import plot_op_curve, plot_op_curve_stat, plot_op_pareto, plot_sa
+from .plot import plot_op_curve, plot_op_curve_stat, plot_op_pareto, plot_sa, \
+                  plot_surrogate, plot_infer_trace, plot_infer_stat, plot_infer_stat_combined
 __all__=[
     'Scaler',
     'MinMaxScaler',
@@ -20,5 +21,9 @@
     'plot_op_curve',
     'plot_op_curve_stat',
     'plot_op_pareto',
-    'plot_sa'
+    'plot_sa',
+    'plot_surrogate',
+    'plot_infer_trace',
+    'plot_infer_stat',
+    'plot_infer_stat_combined',
 ]

UQPyL/util/metric.py

@@ -20,7 +20,7 @@ def mse(true_Y: np.ndarray, pre_Y: np.ndarray) -> np.ndarray:
     """
     Mean square error
     """
-    return np.mean(np.square(true_Y-pre_Y), axis=0)
+    return np.mean(np.square(true_Y - pre_Y), axis=0)
 
 def rank_score(true_Y: np.ndarray, pre_Y: np.ndarray) -> np.ndarray:
     """