使用 sklearn 的简单逻辑回归正在生成 IndexError

Question

我正在尝试使用数据集（CSV 文件：https://drive.google.com/file/d/1EbW22jvbnI0i8JnHC5TqiWemQpbV1zOd/view?usp=sharing）生成一个简单的逻辑回归模型。尝试生成分类图时，产生了回溯错误，原因一直不清楚。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler 
Titanic_data = pd.read_csv('Titanic_noNAN.csv')
x = pd.DataFrame(Titanic_data['Age'])
y = pd.DataFrame(Titanic_data['Survived'])
xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size = 0.25, random_state = 0) 

sc_x = StandardScaler() 
xtrain = sc_x.fit_transform(xtrain)  
xtest = sc_x.transform(xtest)

from matplotlib.colors import ListedColormap 
X_set, y_set = xtest, ytest

X1, X2 = np.meshgrid(np.arange(start = X_set[:, 0].min() - 1,  
stop = X_set[:, 0].max() + 1, step = 0.01), 
np.arange(start = X_set[:, 1].min() - 1,  
stop = X_set[:, 1].max() + 1, step = 0.01)) 

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-253-7e103a17b07d> in <module>
4 X1, X2 = np.meshgrid(np.arange(start = X_set[:, 0].min() - 1,  
5                                stop = X_set[:, 0].max() + 1, step = 0.01), 
----> 6                      np.arange(start = X_set[:, 1].min() - 1,  
7                                stop = X_set[:, 1].max() + 1, step = 0.01)) 
8 

IndexError: index 1 is out of bounds for axis 1 with size 1

Answer 1

你想画什么？ X_set 是您的 xtest 预测，仅是 Age 值，因此 X_set 只有 1 列...