Kriging(克里金)算法 pyhton 基于pykriging包实现

Kriging算法

一种空间插值算法，可以根据数据拟合一个曲面，可以用于预测数值，帮助颜色插值等。
pykriging官网

数据

X_train = [] # position - 位置，里面的内容是[x, y]
y_train = [] # value - 权重，对应上面每一个位置的权重

Kriging模型训练

kriging_model = kriging(X_train, y_train, name='simple')
# 选择ga作为激活函数，还有别的都可
kriging_model.train(optimizer='ga')

预测

# 单点预测
# 把[x, y]换成数值坐标即可得到预测值
kriging_model.predict([x, y])

模型训练结果可视化查看

kriging.plot()