5장 트리 알고리즘 223페이지
----------------
import pandas as pd
from sklearn.model_selection import train_test_split
#wine.describe()
data = wine[['alcohol', 'sugar', 'pH']].to_numpy()
target = wine['class'].to_numpy
train_input, test_input, train_target, test_target = train_test_split(data, target, test_size=0.2, random_state=42)
print(train_input.shape, test_input.shape)
-------------------------------------------
위의 코드를 실행했는데 아래와 같은 오류 메세지가 나옵니다.
---------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-13-12dce3f42a43> in <module>
10 target = wine['class'].to_numpy
11
---> 12 train_input, test_input, train_target, test_target = train_test_split(data, target, test_size=0.2, random_state=42)
13
14 print(train_input.shape, test_input.shape)
4 frames
/usr/local/lib/python3.8/dist-packages/sklearn/utils/validation.py in _num_samples(x)
267 if hasattr(x, "shape") and x.shape is not None:
268 if len(x.shape) == 0:
--> 269 raise TypeError(
270 "Singleton array %r cannot be considered a valid collection." % x
271 )
---------------------------------------
아무리 봐도 오류 원인을 알 수가 없어서 도움 부탁드립니다.