Сохранение прогнозируемых значений классификатора в таблице Excel, python scklearn

Question

Используя sklearn, я предсказал значения.Я хочу сохранить эти прогнозные значения в новый файл Excel вместе с их уникальным идентификатором.

from sklearn.ensemble import AdaBoostClassifier as ABC
from sklearn.tree import DecisionTreeClassifier as DTC
from sklearn.svm import SVC
from sklearn.ensemble import RandomForestClassifier as RFC
from sklearn import linear_model
from sklearn.neural_network import MLPClassifier
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score

# Initialize the models

#adaboostclassifier
clf_A = ABC(random_state = 1)
 clf_A.fit(X_train,y_train)
pred_A=clf_A.predict(X_test)

Я хочу сохранить pred_A и ['SSL'] в файле Xtrain в новый файл csv.Любые предложения о том, как я могу это сделать.

Спасибо

Answer 1

Самый простой способ - что-то вроде этого:

clf_A = ABC(random_state = 1)
clf_A.fit(X_train, y_train)

pred_A = clf_A.predict(X_test)

resultingDF = pd.DataFrame()# you create new dataframe
resultingDF['predictions'] = pred_A# you create column with values
resultingDF['SSL'] = ...

Полагаю, вы берете значения SSL из X_test.Таким образом, если ваш X_test представляет собой pandas dataframe, он будет:

resultingDF['SSL'] = X_test['SSL'].values

Если ваш X_test уже является 2D-массивом, вы используете индекс (для получения столбца) (# 1) или сохраняете (ранее) столбец SSL оттудагде-нибудь и используйте его сейчас (# 2) или вы берете X_test и снова создаете из него фрейм данных, а затем идете с кодом выше (это # ​​3).