解决only integer scalar arrays can be converted to a scalar index

Traceback (most recent call last):
  File "E:/Code/PyCharm/SaltedFish/咸鱼12/K_Fold.py", line 71, in <module>
    x_ = x_train[train]
TypeError: only integer scalar arrays can be converted to a scalar index

问题原因：
制作训练数据我使用的是二维列表，这就会导致在进行样本切片出现问题

解决办法：
使用numpy将二维列表转化成array类型

data = np.array(data_set)