时间戳

首页 > 解决方案 > 如何将预测值输出到 Excel 文件中？

python - 如何将预测值输出到 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)

标签： pythonpandassklearn-pandas

解决方案

你可以这样做：

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)

推荐阅读