Note
Go to the end to download the full example code.
Kriging : quickstart¶
Abstract¶
In this example, we create a Kriging metamodel for a function which has scalar real inputs and outputs. We show how to create the learning and the validation samples. We show how to create the Kriging metamodel by choosing a trend and a covariance model. Finally, we compute the predicted Kriging confidence interval using the conditional variance.
Introduction¶
We consider the sine function:
for any .
We want to create a metamodel of this function. This is why we create a sample of observations of the function:
for , where is the ith input and is the corresponding output.
We consider the seven following inputs :
1 
2 
3 
4 
5 
6 
7 


1 
3 
4 
6 
7.9 
11 
11.5 
We are going to consider a Kriging metamodel with:
a constant trend,
a Matern covariance model.
Creation of the metamodel¶
We begin by defining the function g as a symbolic function. Then we define the x_train variable which contains the inputs of the design of experiments of the training step. Then we compute the y_train corresponding outputs. The variable n_train is the size of the training design of experiments.
import openturns as ot
from openturns import viewer
from matplotlib import pylab as plt
ot.Log.Show(ot.Log.NONE)
g = ot.SymbolicFunction(["x"], ["sin(x)"])
x_train = ot.Sample([[x] for x in [1.0, 3.0, 4.0, 6.0, 7.9, 11.0, 11.5]])
y_train = g(x_train)
n_train = x_train.getSize()
n_train
7
In order to compare the function and its metamodel, we use a test (i.e. validation) design of experiments made of a regular grid of 100 points from 0 to 12. Then we convert this grid into a Sample and we compute the outputs of the function on this sample.
xmin = 0.0
xmax = 12.0
n_test = 100
step = (xmax  xmin) / (n_test  1)
myRegularGrid = ot.RegularGrid(xmin, step, n_test)
x_test = myRegularGrid.getVertices()
y_test = g(x_test)
In order to observe the function and the location of the points in the input design of experiments, we define the following functions which plots the data.
def plot_data_train(x_train, y_train):
"""Plot the data (x_train,y_train) as a Cloud, in red"""
graph_train = ot.Cloud(x_train, y_train)
graph_train.setColor("red")
graph_train.setLegend("Data")
return graph_train
def plot_data_test(x_test, y_test):
"""Plot the data (x_test,y_test) as a Curve, in dashed black"""
graphF = ot.Curve(x_test, y_test)
graphF.setLegend("Exact")
graphF.setColor("black")
graphF.setLineStyle("dashed")
return graphF
graph = ot.Graph("test and train", "", "", True, "")
graph.add(plot_data_test(x_test, y_test))
graph.add(plot_data_train(x_train, y_train))
graph.setAxes(True)
graph.setXTitle("X")
graph.setYTitle("Y")
graph.setLegendPosition("upper right")
view = viewer.View(graph)
We use the ConstantBasisFactory class to define the trend and the MaternModel class to define the covariance model. This Matérn model is based on the regularity parameter .
dimension = 1
basis = ot.ConstantBasisFactory(dimension).build()
# basis = ot.LinearBasisFactory(dimension).build()
basis = ot.QuadraticBasisFactory(dimension).build()
covarianceModel = ot.MaternModel([1.0] * dimension, 1.5)
algo = ot.KrigingAlgorithm(x_train, y_train, covarianceModel, basis)
algo.run()
result = algo.getResult()
print(result)
KrigingResult(covariance models=MaternModel(scale=[1.0568], amplitude=[0.872754], nu=1.5), covariance coefficients=0 : [ 0.369718 ]
1 : [ 0.473109 ]
2 : [ 1.47872 ]
3 : [ 0.45338 ]
4 : [ 1.71228 ]
5 : [ 0.967332 ]
6 : [ 0.344328 ], basis=Basis( [class=LinearEvaluation name=Unnamed center=[0] constant=[1] linear=[[ 0 ]],class=LinearEvaluation name=Unnamed center=[0] constant=[0] linear=[[ 1 ]],QuadraticEvaluation
center :
[0]
constant :
[0]
linear :
[[ 0 ]]
quadratic :
sheet #0
[[ 2 ]]
] ), trend coefficients=[0.667411,0.117016,0.000810156])
We observe that the scale and amplitude hyperparameters have been optimized by the run method. Then we get the metamodel with getMetaModel and evaluate the outputs of the metamodel on the test design of experiments.
krigeageMM = result.getMetaModel()
y_test_MM = krigeageMM(x_test)
The following function plots the Kriging data.
def plot_data_kriging(x_test, y_test_MM):
"""Plots (x_test,y_test_MM) from the metamodel as a Curve, in blue"""
graphK = ot.Curve(x_test, y_test_MM)
graphK.setColor("blue")
graphK.setLegend("Kriging")
return graphK
graph = ot.Graph("", "", "", True, "")
graph.add(plot_data_test(x_test, y_test))
graph.add(plot_data_train(x_train, y_train))
graph.add(plot_data_kriging(x_test, y_test_MM))
graph.setAxes(True)
graph.setXTitle("X")
graph.setYTitle("Y")
graph.setLegendPosition("upper right")
view = viewer.View(graph)
We see that the Kriging metamodel is interpolating. This is what is meant by conditioning a gaussian process.
We see that, when the sine function has a strong curvature between two points which are separated by a large distance (e.g. between and ), then the Kriging metamodel is not close to the function . However, when the training points are close (e.g. between and ) or when the function is nearly linear (e.g. between and ), then the Kriging metamodel is quite accurate.
Compute confidence bounds¶
In order to assess the quality of the metamodel, we can estimate the Kriging variance and compute a 95% confidence interval associated with the conditioned gaussian process.
We begin by defining the alpha variable containing the complementary of the confidence level than we want to compute. Then we compute the quantile of the gaussian distribution corresponding to 1alpha/2. Therefore, the confidence interval is:
alpha = 0.05
def computeQuantileAlpha(alpha):
bilateralCI = ot.Normal().computeBilateralConfidenceInterval(1  alpha)
return bilateralCI.getUpperBound()[0]
quantileAlpha = computeQuantileAlpha(alpha)
print("alpha=%f" % (alpha))
print("Quantile alpha=%f" % (quantileAlpha))
alpha=0.050000
Quantile alpha=1.959964
In order to compute the Kriging error, we can consider the conditional variance. The getConditionalCovariance method returns the covariance matrix covGrid evaluated at each points in the given sample. Then we can use the diagonal coefficients in order to get the marginal conditional Kriging variance. Since this is a variance, we use the square root in order to compute the standard deviation. However, some coefficients in the diagonal are very close to zero and nonpositive, which leads to an exception of the sqrt function. This is why we add an epsilon on the diagonal (nugget factor), which prevents this issue.
sqrt = ot.SymbolicFunction(["x"], ["sqrt(x)"])
epsilon = ot.Sample(n_test, [1.0e8])
conditionalVariance = result.getConditionalMarginalVariance(x_test) + epsilon
conditionalSigma = sqrt(conditionalVariance)
The following figure presents the conditional standard deviation depending on .
graph = ot.Graph(
"Conditional standard deviation", "x", "Conditional standard deviation", True, ""
)
curve = ot.Curve(x_test, conditionalSigma)
graph.add(curve)
view = viewer.View(graph)
We now compute the bounds of the confidence interval. For this purpose we define a small function computeBoundsConfidenceInterval :
def computeBoundsConfidenceInterval(quantileAlpha):
dataLower = [
[y_test_MM[i, 0]  quantileAlpha * conditionalSigma[i, 0]]
for i in range(n_test)
]
dataUpper = [
[y_test_MM[i, 0] + quantileAlpha * conditionalSigma[i, 0]]
for i in range(n_test)
]
dataLower = ot.Sample(dataLower)
dataUpper = ot.Sample(dataUpper)
return dataLower, dataUpper
We define two small lists to draw three different confidence intervals (defined by the alpha value) :
alphas = [0.05, 0.1, 0.2]
# three different green colors defined by HSV values
mycolors = [[120, 1.0, 1.0], [120, 1.0, 0.75], [120, 1.0, 0.5]]
We are ready to display all the previous information and the three confidence intervals we want.
sphinx_gallery_thumbnail_number = 4
graph = ot.Graph("", "", "", True, "")
graph.add(plot_data_test(x_test, y_test))
graph.add(plot_data_train(x_train, y_train))
graph.add(plot_data_kriging(x_test, y_test_MM))
# Now we loop over the different values :
for idx, v in enumerate(alphas):
quantileAlpha = computeQuantileAlpha(v)
vLow, vUp = computeBoundsConfidenceInterval(quantileAlpha)
boundsPoly = ot.Polygon.FillBetween(x_test, vLow, vUp)
boundsPoly.setColor(
ot.Drawable.ConvertFromHSV(mycolors[idx][0], mycolors[idx][1], mycolors[idx][2])
)
boundsPoly.setLegend(" %d%% bounds" % ((1.0  v) * 100))
graph.add(boundsPoly)
graph.setAxes(True)
graph.setXTitle("X")
graph.setYTitle("Y")
graph.setLegendPosition("upper right")
view = viewer.View(graph)
We see that the confidence intervals are small in the regions where two consecutive training points are close to each other (e.g. between and ) and large when the two points are not (e.g. between and ) or when the curvature of the function is large (between and ).
plt.show()
References¶
Metamodeling with Gaussian processes, Bertrand Iooss, EDF R&D, 2014, www.gdrmascotnum.fr/media/sssamo14_iooss.pdf