python-logistische Regression, speichern vorhergesagte Wahrscheinlichkeiten und Vorhersagen zu csv

Question

Ich bin neu in der Datenanalyse und Python. Ich habe meistens verfolgt die logistische Regression in diesem Link zeigte auf titanic Überlebenden:

http://hamelg.blogspot.ca/2015/11/python-for-data-analysis-part-28.html

ich aber meinen eigenen nicht titanic Dataset verwenden. Ich bin am Ende des Beispiels, wo ich die Ergebnisse in eine CSV-Datei exportieren möchte. Ich habe jedoch eine kleine Änderung vorgenommen, die mich feststecken lässt. Zusätzlich zur Vorhersage habe ich explizit die vorhergesagten Wahrscheinlichkeiten generiert, die ich auch in die CSV-Datei exportieren möchte.

test_probs=log_model.predict_proba(X=test_features) 
print(test_probs) 


# Create a submission for Kaggle 
submission = pd.DataFrame({"AccountNumber":titanic_test["AccountNumber"], 
"PolarPredict":test_preds,**"probabilities":test_probs** }) 

Dies ist die Botschaft i erhalten: Ausnahme: Daten

hier ist der ursprüngliche Code aus dem Tutorial 1-dimensional sein muss: # Stellen Testset Prognosen test_preds = log_model.predict (X = test_features)

# Create a submission for Kaggle 
submission = pd.DataFrame({"PassengerId":titanic_test["PassengerId"], 
         "Survived":test_preds}) 

# Save submission to CSV 
submission.to_csv("tutorial_logreg_submission.csv", 
       index=False) 

Wie kann ich exportieren Sie die Vorhersage, Wahrscheinlichkeiten und die "ID" in eine cSV-Datei?

Answer 1

können Sie versuchen, sie in die Liste Umwandlung

submission = pd.DataFrame({ 
    "AccountNumber":list(titanic_test["AccountNumber"]), 
    "PolarPredict":list(test_preds), 
    "probabilities":list(test_probs) 
    }) 

submission.to_csv("tutorial_logreg_submission.csv", 
      index=False)