Как сохранить исходные индексы в массивном массиве, выводимом из функции Преди

Question

Я обучил модель xgb и вызвал функцию прогнозирования в наборе тестовых данных

xg_reg.fit(X_train, y_train)

preds = xg_reg.predict(X_test)

Однако результирующая переменная preds представляет собой массив numpy, где индексы сбрасываются на 0-101

Так как мой y_test является фреймом данных pandas со случайными индексами от деления его на train_test_split from (sklearn.model_selection) Я не могу на самом деле отобразить или добавить полученные прогнозируемые значения в тестовый фрейм данных.

Есть лиспособ сохранить исходную индексацию в прогнозируемом массиве?

Answer 1

возьмите y_test.values() и сравните их непосредственно с pred, трактуйте их как массивы и проводите сравнения, прогнозы определенно будут соответствовать фактическим данным, индексы в этом случае ничего не определяют, если вы думаете, что естьнесоответствие между preds и y_test из-за низкой точности, это определенно ошибка модели или данных или что-то еще, но прогнозы соответствуют y_test