Assertion Error解決法について

Question

以下のコードでAssertion Errorを回避する方法を教えて頂けないでしょうか？

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=2)
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

ちなみにX.shapeは(248, 4)
Y.shapeは(248,)　
です。

Comments

[退会済みユーザー]

AssertionErrorの続きに「何行のどこがどういう風にダメ」、というようなメッセージはありましたか？ありましたらそこもご記入ください。

[Nats91115796]

AssertionErrorの続きにメッセージはありませんでした。

Answer 1

■assertの働き
元のコードの全体がないので何とも言えませんが、
assert Y.ndim == 2は、Y.ndimが2でなければ強制的にエラーとみなして終了させるという意味の命令になります。

なので、私のコメントの

AssertionErrorの続きに「何行のどこがどういう風にダメ」、というようなメッセージはありましたか？ありましたらそこもご記入ください。

はかなりナンセンスなコメントでした。assertを使った時点で通常はプログラム側からその他の情報はつかないためです。

■解決策
上記の通り、Y.ndimを2にするか、assertの部分の数値を1にすればよさそうです。

Answer 2

Y = train_df[['flower']]