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()