n-dimensional Kriging

[Vincy-xiao]

Hi, I am using pyKriging to train a set of input(100 rows and 3 columns) and output(100 rows and 400 columns) data, but it shows that there is a problem: index 1 is out of bounds for axis 0 with size 1.
Below is the simple code I wrote:

import math
import random
import numpy as np
import matplotlib.pyplot as plt
from pyKriging.krige import kriging
import scipy.io as scio

path_1 = 'hb114_00_bywz.mat'
data_1 = scio.loadmat(path_1)

X = data_1['params'] #100 rows and 3 columns
Y = data_1['Cp_M'] #100 rows and 400 columns
X = np.mat(X)
Y = np.mat(Y)
k0 = kriging(X, Y)
k0.train()
a = [0.1065, 0.062, 0.353]
YX = k0.predict(a)

These codes can't run, hope to get help, thanks in advance.

[capaulson]

There should only be one column in Y. If you have 400 columns in Y, you would need 400 models to predict each column.