I am trying to load and split my data however i get TypeError: 'only integer scalar arrays can be converted to a scalar index'

Question

进口

import numpy as np
from sklearn.linear_model import LogisticRegression # Architecture model
from sklearn.datasets import _olivetti_faces # Dataset
from sklearn.model_selection import train_test_split

### Loading and splitting the data
data = np.load("/content/Face_Recognition/olivetti_faces.npy")
Xtrain, Xtest, Ytrain, Ytest = train_test_split(data.data, test_size=0.3, random_state=0)
#Error to show if x and y rows mismatch
print(X.shape)
print(y.shape)
if X.shape[0] != y.shape[0]:

  print("X and y rows are mismatched, check dataset again")

Answer 1

您似乎使用的是过时版本的数据集。您会在此处看到更新后的 link。您拥有的代码将不起作用，因为您要求拆分数据和目标，而您只提供目标。

像这样的东西应该可以工作（在 sklearn 0.24.1 上）：

from sklearn.datasets import fetch_olivetti_faces

data = fetch_olivetti_faces()
Xtrain, Xtest, Ytrain, Ytest = train_test_split(data.data, data.target, test_size=0.3, random_state=0)

I am trying to load and split my data however i get TypeError: 'only integer scalar arrays can be converted to a scalar index'

I am trying to load and split my data however i get TypeError: 'only integer scalar arrays can be converted to a scalar index'

python

scikit-learn

jupyter-notebook

进口