2016-01-18 17 views
7

Sonuçları bir sklearn regresyon modelinden varibla tahminine saklıyorum.Öngörme sonuçlarının CSV'ye kaydedilmesi

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

Ama: tahmini çıkış

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

değerleri bu

[ 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] 
Sonra yerel bir CSV dosyasına sonuçları kaydetmek için to_csv işlevini kullanmaya çalışıyorum

benziyor Ben dönene hatadır:

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

Pandas/Numpy/SKlearn kullanıyorum. Temel düzeltme hakkında bir fikrin var mı?

cevap

6

Pandaları kullanabilirsiniz. Dediğimiz gibi, numpy dizilerinde bir to_csv işlevi yoktur. Değerlerinizin satır veya sütun benzeri olmasını istiyorsanız, ".T" ifadesini ekleyiniz. .

6

Sen numpy.savetxt işlevini kullanabilirsiniz:

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

ve CSV numpy.genfromtxt fonksiyonunu kullanabilirsiniz dosya yüklemek için:

numpy.genfromtxt('C:/localpath/test.csv', delimiter=',') 
İlgili konular