python - Python：方法 .as_matrix 将在未来版本中删除。改用 .values

Question

我有以下代码

train_X, test_X, train_y, test_y = train_test_split(X.as_matrix(), y.as_matrix(), test_size=0.25)

其中X是一个 DataFrame 并且y是一个系列。调用上面的函数时，我收到以下警告：

/opt/conda/lib/python3.6/site-packages/ipykernel_launcher.py:1：FutureWarning：方法 .as_matrix 将在未来版本中删除。请改用 .values。

"""启动 IPython 内核的入口点。

然后我尝试.values按照警告中的说明进行更改：

train_X, test_X, train_y, test_y = train_test_split(X.values(), y.values(), test_size=0.25)

但我收到以下错误：

TypeError Traceback (last last call last) in () ----> 1 train_X, test_X, train_y, test_y = train_test_split(X.values(), y.values(), test_size=0.25)

TypeError：“numpy.ndarray”对象不可调用

我该如何解决这个问题？

标签： pythonscikit-learn

Answer 1

它应该是：

train_X, test_X, train_y, test_y = train_test_split(X.values, y.values, test_size=0.25)

看到这个。