Из большого набора данных я пытаюсь экспортировать два столбца в csv: 'row_id' и 'accept'. «Принятый» содержит 0 или 1 на строку. Это результат модели машинного обучения.
Однако, когда я проверяю CSV, все 0 исчезли. Другими словами, если значение «принятого» столбца в моих данных равно 0, в csv оно пустое. Как я могу это исправить?
Я попытался определить тип данных следующим образом:
my_submission=pd.DataFrame({'row_id':testvalues.row_id,'accepted': solution2}).astype('str')
Но вывод был тот же.
Мой текущий код:
solution2=gbr.predict(testvalues1)
my_submission=pd.DataFrame({'row_id':testvalues.row_id,'accepted': solution2})
my_submission.to_csv('GradientBoosterClassifierSolution.csv', index=False)
Текущие результаты:
row_id, accepted
0,1
1
2
3,1
4
Ожидаемые результаты:
row_id, accepted
0,1
1,0
2,0
3,1
4,0