如何在excel中将预测值输出为字符串？

Question

所以我能够将我的预测数值输出到 excel 文件中，但我想知道是否可以代替数值，而是实际导出字符串。

目前是这样的，

Column 1	Answer Key	Predicted
Something Something	Cars	3

与其返回 3，我希望将 3 替换为其关联的实际字符串（例如，idk Truck）。

到目前为止，这是我的代码，我知道我必须弄乱代码的导出部分，但我似乎无法弄清楚这一点。

texts = df['without_Tags'].astype('str')

vector = TfidfVectorizer(ngram_range=(1, 2), min_df = 2, max_df = .95)
X = vector.fit_transform(texts) #features

LE = LabelEncoder()
df['tower_values'] = LE.fit_transform(df['Tower'])
y = df['tower_values'].values

print(X.shape)
print(y.shape)

lsa = TruncatedSVD (n_components=100, n_iter=10, random_state=3)

X = lsa.fit_transform(X)
print(X.shape)

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = .3, shuffle = True, stratify = y, random_state = 3)

SG = SGDClassifier(random_state=3, loss='log')
SG.fit(X_train, y_train)
y_pred = SG.predict(X_test)

print("SG model accuracy:", accuracy_score(y_test, y_pred))
print("SG model Recall:", recall_score(y_test, y_pred, average="macro"))
print("SG model Precision:", precision_score(y_test, y_pred, average="macro"))
print("SG model F1 Score:", f1_score(y_test, y_pred, average="macro"))


y_pred = pd.DataFrame(y_pred, columns=['predictions']).to_csv('prediction.csv')

final = pd.read_csv('prediction.csv')
final['pre'] = y_pred
df.to_csv('prediction.csv')

Answer 1

尝试使用LabelEncoder()

的逆变换方法

y_pred = LE.inverse_transform(y_pred)

如何在excel中将预测值输出为字符串？

How to output predicted values as a string in excel?

python

pandas

sklearn-pandas