Простая регрессия logisti c с sklearn генерирует IndexError

Question

Я пытаюсь создать простую логистическую модель c регрессии с использованием набора данных (файл 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, которые являются только значениями возраста, поэтому X_set имеет только 1 столбец ...