capaulson / pyKriging

Welcome to the User Friendly Python Kriging Toolbox!
http://www.pykriging.com
MIT License
386 stars 114 forks source link

n-dimensional Kriging #41

Open Vincy-xiao opened 3 years ago

Vincy-xiao commented 3 years ago

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 commented 3 years ago

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