Tensorflow classifier.predict寫入csv

Question

我沒有使用Tensorflow一段時間，當我更新它似乎破壞了我的舊代碼，因爲許多舊功能已棄用。我固定的他們用新的代碼和這一切似乎除了在運行的時候我寫出來的結果：

y_predicted = classifier.predict(X_test) 

有一個作爲迭代選項以及 - 我不認爲我需要的。

我用寫出來用預測的結果：

pandas.DataFrame(y_predicted).to_csv(/dir/) 

，但現在我得到，並不是所有的元素可以被轉換成字符串類型的錯誤。 y_predicted中有沒有一個類，我想要調用而不是整個事情？

Answer 1

不管怎麼說，我找到了解決辦法使用的，而不是一個熊貓數據幀np.array：

result = np.asarray(y_predicted) 
formatInt = result.astype(np.int) 
np.savetxt("dir",formatInt,delimiter=",") 

Answer 2

您也可以嘗試，

df = pandas.DataFrame({'Prediction':list(y_predicted)}) 
df.to_csv('filename.csv')