GPyによるモデリング

Question

Pythonによる機械学習初心者です。
おそらく基本的なところでつまずいていますが、以下のエラーの解決策を教えて頂けないでしょうか？

import pandas as pd
import GPy

get_train_df = pd.read_excel('dataset.xlsx',sheet_name='Dataset')
train_df = get_train_df.set_index('case')

X = train_df[['red', 'blue', 'yellow', 'green']]
Y = train_df['flower']
kernel = GPy.kern.RBF(input_dim=1)
model = GPy.models.GPRegression(X, Y)
model.optimize()

以下がエラーコードです。

  File "C:\Users\Na\Anaconda3\lib\site-packages\GPy\core\gp.py", line 48, in __init__
    assert Y.ndim == 2

AssertionError

Comments

[WathMorks]

X.shape
Y.shape
を確認しましょう。

[Nats91115796]

print(X.shape)
print(Y.shape)

#(248, 4)
#(248,)

です。

Answer 1

Y = train_df[['flower']]

Answer 2

kernel = GPy.kern.RBF(input_dim=4)

Comments

[Nats91115796]

input_dim=1,2,3,4　全て試してみましたが全て同じエラーコードが出てしまいます。次元に関する記述はこの点のみですが、他に問題があるのでしょうか、、？

[WathMorks]

X = train_df[['red', 'blue', 'yellow', 'green']].values
Y = train_df['flower'].values
として試してみてください。