mglearn.plots.plot_2d_separator()のコードで、 unhashable type: 'slice' というエラーが表示される。

Question

import pandas as pd
import numpy as np
import mglearn
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.svm import LinearSVC

csv_titanic = pd.read_csv("titanic_train.csv")

#　読み取ったファイルを欠陥値処理やダミー変数化をして前処理を行った（中略）

X = csv_titanic[["Age","Fare"]]
y = csv_titanic["Survived"]　#0と1のラベル

X_train,X_test,y_train, y_test=train_test_split(X, y, stratify=y )

svm = LinearSVC().fit(X_train,y_train)
plt.figure()
mglearn.plots.plot_2d_separator(svm,X)


TypeError: unhashable type: 'slice'

mglearn.plots.plot_2d_separator(svm,X)を上のコードで使用すると、
unhashable type: 'slice'
というエラーが表示されます。

上のコードでX,yともにsliceにはなっていないと思うのですが何が原因なのでしょうか？

よろしくお願いします。

Answer 1

原因はわからないのですが...
svm,Xをplotさせる前にnp.arrayに変換すれば解決するかと。

Comments

[1mzmk]

ご回答していただきありがとうございます。

np.arrayに変換したのですが
AttributeError: 'numpy.ndarray' object has no attribute 'predict_proba'
というエラーが表示され解決できませんでした。

[1mzmk]

すいません、Xだけ配列変換して実行すれば、問題なくコードが通りました。

ありがとうございました。