В приведенном ниже коде второй вызов XGBoosterPredict перезаписывает train_out_pb,
проблема, наверное, очень тривиальная, но я не вижу ее.
вывод отладчика после первого вызова:
вывод отладчика после 2-го вызова:
Код:
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();
}
Не идеально, но решение, тем не менее, заключается в том, чтобы сохранить перезаписанную переменную в другой переменной
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);
Других решений пока нет …