IndexError：目標 5 越界

Question

我正在嘗試訓練 ANN 但得到

IndexError：目標 5 越界

我認為問題出在我拆分數據的這部分代碼中：

from sklearn.datasets import fetch_california_housing
california = fetch_california_housing()
data = pd.DataFrame(california.data)
data.columns = california.feature_names
data['Price'] = california.target

X = data.iloc[:, 0:8]
y = data.iloc[:, 8]

我在這里做錯了什么嗎？

Answer 1

添加新列然后按列索引進行選擇很容易出錯。

在scikit-learn>=0.23.0中， fetch_california_housing已經可以使用as_frame參數返回 dataframe。

如果您需要數據框，您的代碼結構應如下所示：

from sklearn.datasets import fetch_california_housing

california = fetch_california_housing(as_frame=True)

X = california.data
y = california.target

Answer 2

我希望這可能有所幫助：

from sklearn.datasets import fetch_california_housing

# dataframe
df = fetch_california_housing(as_frame=True).frame

dataset = fetch_california_housing()
X, y = dataset.data, dataset.target
features = dataset.feature_names
features

['MedInc',
 'HouseAge',
 'AveRooms',
 'AveBedrms',
 'Population',
 'AveOccup',
 'Latitude',
 'Longitude']