如何解决如何将预测值输出到 Excel 文件中?
scikit-learn 的新手,我想获取预测值并将其转换回文本并将其输出到 excel 文件中。
项目的设置方式是获取一行字符串并预测该列是否属于某个类别(大约有 5 个类别)。
描述 | 实际答案 | 预测 |
---|---|---|
一些每行长度随机的字符串 | 汽车 | 卡车 |
我想要像上面看到的那样输出 excel 文件。我不想输出数值预测结果。我想输出实际的文本本身。
谁能帮助我如何做到这一点?
这是我目前的代码:
X = df['without_Tags']
Y = df['Tower']
tokens = Tokenizer()
VectorX = tokens.texts_to_sequences(df['without_Tags'].values)
VectorX = pad_sequences(VectorX,maxlen=200)
VectorY = pd.get_dummies(df['Tower'])
X_train,X_test,y_train,y_test = train_test_split(VectorX,VectorY,test_size=0.20,random_state=0)
# Model Creation
from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(X_train,y_train)
y_pred = knn.predict(X_test)
解决方法
你可以这样做:
import pandas as pd
CSV = pd.DataFrame({
"Prediction": y_pred
})
CSV.to_csv("prediction.csv",index=False)
该文件将命名为“prediction.csv”并保存在您的源代码文件目录中。
更新:
import pandas as pd
csv = pd.DataFrame(y_pred,columns=["1st","2nd","3rd","4th","5th"])
csv.to_csv("pred.csv",index=False)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。