XGBoosterPredict second appel écrase le précédent

Question

Dans le code ci-dessous, le deuxième appel à XGBoosterPredict remplace le fichier train_out_pb, est probablement très trivial mais je ne le vois pas.

sortie du débogueur après 1.st appel:

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

sortie du débogueur après 2ème appel:

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

code:

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); 

extrait de 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(); 
} 

Answer 1

Pas parfait, mais une solution est néanmoins de stocker variabl écrasé e dans une autre variable

 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);