dans=pd.read_csv('ans.csv', header=0)
dataa=dans
for x in dataa['ideas']:
token=tokenizer.texts_to_matrix([x], mode='tfidf')
c=model.predict(np.array(token))
#cc=model.predict_classes(token)
cc=np.argmax(c,axis=1)
xc=encoder.inverse_transform(cc)
print((xc))
我想把xc的结果保存在csv文件中。它只是保存xc中的最后一个预测，而不是所有预测。
dans=pd.read_csv('ans.csv', header=0)
results = dans['A_COLUMN_NAME']
results['Prediction'] = 0
idx = 0
dataa=dans
for x in dataa['ideas']:
token=tokenizer.texts_to_matrix([x], mode='tfidf')
c=model.predict(np.array(token))
#cc=model.predict_classes(token)
cc=np.argmax(c,axis=1)
xc=encoder.inverse_transform(cc)
results['Prediction'].iloc[idx] = xc
idx += 1
print((xc))
results.to_csv('Results.csv')
只是在这里超级懒惰->results = dans['A_COLUMN_NAME']
你可以试试更经典的results = pd.DataFrame(index=dans.index)：(