Speichern Vorhersage-Ergebnisse zu CSV

Ich bin die Speicherung der Ergebnisse von einem sklearn regression-Modell, um die varibla Vorhersage.

prediction = regressor.predict(data[['X']])
print(prediction)

Die Werte der Vorhersage Ausgabe sieht wie folgt aus

[ 266.77832991  201.06347505  446.00066136  499.76736079  295.15519906
  214.50514991  422.1043505   531.13126879  287.68760191  201.06347505
  402.68859792  478.85808879  286.19408248  192.10235848]

Bin ich dann versuchen, zu verwenden die to_csv-Funktion zum speichern der Ergebnisse in eine lokale CSV-Datei:

prediction.to_csv('C:/localpath/test.csv')

Aber die Fehler, die ich zurück bekomme ist:

AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'

Ich bin mit Pandas/Numpy/SKlearn. Keine Ahnung über die grundlegenden beheben?

InformationsquelleAutor ZJAY | 2016-01-18
  1. 20

    Können Sie pandas.
    Wie es heißt, numpy-arrays, die nicht über eine to_csv Funktion.

    import numpy as np
import pandas as pd
prediction = pd.DataFrame(predictions, columns=['predictions']).to_csv('prediction.csv')

    hinzufügen ".T" wenn Sie wollen entweder Ihre Werte in der Zeile oder Spalte-wie.

    • Wenn ich Zusammenführen möchten, die mit einer einzigartigen Kennung von X_test ("id" - Spalte, nicht der index), wird die Vorhersage richtig passen jeder Zeile? in: output=pd.DataFrame(data={"id":X_test["id"],"Prediction":y_pred}) output.to_csv(path_or_buf="..\\output\\results.csv",index=False,quoting=3,sep=';')
    InformationsquelleAutor DavidK
  2. 10

    Können Sie die numpy.savetxt Funktion:

    numpy.savetxt('C:/localpath/test.csv',prediction, ,delimiter=',')

    und laden Sie eine CSV-Datei, die Sie verwenden können numpy.genfromtxt Funktion:

    numpy.genfromtxt('C:/localpath/test.csv', delimiter=',')
    • Ich hatte umzugestalten meine Daten nach dem laden ich.e: "pred_train = np.genfromtxt('kodiert1.csv', delimiter=" ").Umformen(-1, 1)", gibt es nicht eine Möglichkeit zum speichern und laden der Daten, ohne darüber nachzudenken, reshaping it?
    InformationsquelleAutor Ali
Schreibe einen Kommentar