Второй вызов XGBoosterPredict перезаписывает предыдущий

В приведенном ниже коде второй вызов XGBoosterPredict перезаписывает train_out_pb,
проблема, наверное, очень тривиальная, но я не вижу ее.

вывод отладчика после первого вызова:

  • &train_out_pb 0x06d0eb2c {0x0074d060 {0.410961390}} const float * *
  • &val_out_pb 0x06d0eb28 {0xffffffff {???}} const float * *

вывод отладчика после 2-го вызова:

  • &train_out_pb 0x06d0eb2c {0x0074d060 {0.522293866}} const float * *
  • &val_out_pb 0x06d0eb28 {0x0074d060 {0.522293866}} const float * *

Код:

bst_ulong train_pred_len;
const float* train_out_pb = new float[train_num_samples]; //()
//1.st call
ret = XGBoosterPredict(h_booster, XY_train, 0, 0, &train_pred_len, &train_out_pb);

bst_ulong val_pred_len;
const float* val_out_pb = new float[val_num_samples];
//2.nd call
ret = XGBoosterPredict(h_booster, XY_val, 0, 0, &val_pred_len, &val_out_pb);

выдержка из XGBoost c_api.cc:

XGB_DLL int XGBoosterPredict(BoosterHandle handle,
DMatrixHandle dmat,
int option_mask,
unsigned ntree_limit,
xgboost::bst_ulong *len,
const bst_float **out_result)
{
std::vector<bst_float>& preds = XGBAPIThreadLocalStore::Get()->ret_vec_float;
API_BEGIN();
Booster *bst = static_cast<Booster*>(handle);
bst->LazyInit();

bst->learner()->Predict(
static_cast<std::shared_ptr<DMatrix>*>(dmat)->get(),
(option_mask & 1) != 0,
&preds, ntree_limit,
(option_mask & 2) != 0);
*out_result = dmlc::BeginPtr(preds);
*len = static_cast<xgboost::bst_ulong>(preds.size());
API_END();
}

0

Решение

Не идеально, но решение, тем не менее, заключается в том, чтобы сохранить перезаписанную переменную в другой переменной

        const float* train_out_pb;

ret = XGBoosterPredict(h_booster, XY_train, 0, 0, &train_pred_len, &train_out_pb);

std::vector<float> train_out_pb_v(train_out_pb, train_out_pb + train_pred_len);
0

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]