SKlearn Logistische Regression schlechte Eingangs Form ERROR

Question

Wie kann ich diesen Fehler beseitigen ist

from sklearn.linear_model import LogisticRegression 
from sklearn.model_selection import train_test_split 
import matplotlib.pyplot as plt 
import pandas as pd 

df = pd.read_csv("train.csv") 

clean = {"Sex": {"male":1, "female":0}} 
df.replace(clean, inplace = True) 
df["label"] = df['Survived'] 
df = df.drop(["Name","Ticket","Cabin","Embarked","Fare","Parch","Survived"], axis = 1) 
df = df.dropna(axis = 0, how="any") 

X = df.drop(["label"],axis = 1).values 
y = df["label"].values 

X_train , y_train, X_test, y_test = train_test_split(X, y, test_size = 0.7) 

log_reg = LogisticRegression() 
log_reg.fit(X_train, y_train) 
print("Accuracy on test subset: (:.3f)".format(log_reg.score(X_train, y_train))) 

ERROR 
Traceback (most recent call last): 
    File "C:\Users\user\Documents\17\kaggle'\logistic.py", line 20, in <module> 
    log_reg.fit(X_train, y_train) 
    File "C:\Users\user\AppData\Local\Programs\Python\Python36-32\lib\site-packages\sklearn\linear_model\logistic.py", line 1216, in fit 
    order="C") 
    File "C:\Users\user\AppData\Local\Programs\Python\Python36-32\lib\site-packages\sklearn\utils\validation.py", line 547, in check_X_y 
    y = column_or_1d(y, warn=True) 
    File "C:\Users\user\AppData\Local\Programs\Python\Python36-32\lib\site-packages\sklearn\utils\validation.py", line 583, in column_or_1d 
    raise ValueError("bad input shape {0}".format(shape)) 
ValueError: bad input shape (500, 5) 

Answer 1

Der Fehler aufgrund dieser:

X_train , y_train, X_test, y_test = train_test_split(X, y, test_size = 0.7) 

Dies ist nicht das, was die train_test_split kehrt zurück.

Das sollte die tatsächliche Nutzung sein:

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.7) 

train_test_split werden die gespaltet Arrays in der Reihenfolge der gelieferten Daten zurückzukehren. Also wird X in X_train, X_test aufgeteilt und zuerst zurückgegeben, dann wird y als y_train y_test zurückgegeben. Hoffe, das hilft.