Как сохранить значения прогноза в виде строк в моей регрессионной модели logisti c в python?

Question

этот код дает мне значения прогноза как 1 0r 0. Но я хочу напечатать их как «Успех» или «Сбой». Есть ли способ для этого?

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

Answer 1

У меня нет большого опыта работы с pandas, но если вы просто хотите назначить эти значения для этих выходных данных, один из способов - это словарь (опубликованный как я набираю). Вы также можете использовать троичный оператор. Пример:

prediction = pd.DataFrame(predictions2, columns ['predictions']).to_csv('prediction.csv')
result = 'Success' if prediction == 1 else 'Failure'
print(result)

Это может быть дополнительно уменьшено до:

print('Success' if prediction == 1 else 'Failure')

Примечание: Используйте 1, если выходное значение равно int, используйте '1', если строка.

Answer 2

Спасибо. Все решено!

**mydict ={0:'Failure',1:'Success'}
predictions2 = [mydict[i] for i in predictions2]
print(predictions2)
predictions2= pd.DataFrame(predictions2, columns=['predictions']).to_csv('prediction3.csv', index=False)

Answer 3

Все, что вам нужно сделать, это сопоставить 0 и 1 с ошибками и успехами, используя словарь

prediction = pd.DataFrame(predictions2, columns=['predictions'])
prediction['predictions'] = prediction['predictions'].astype(int)
prediction["predictions"] = prediction.predictions.map({0:'Failure', 1:'Success'})
prediction.to_csv('prediction.csv', index=False)

Необходимо убедиться, что тип данных «прогнозы» - int.