preds_to_matrix: Convert vector of predictions to matrix format From line 56...

Description Usage Arguments Value

Description

Convert vector of predictions to matrix format From line 56 at https://github.com/dmlc/xgboost/blob/ef4dcce7372dbc03b5066a614727f2a6dfcbd3bc/src/objective/multiclass_obj.cc, it appears that preds is stored in column-major order with observations in columns and classes/models in rows i.e., preds[k * nclass + i] is prediction for model i at index k

Usage

1
preds_to_matrix(preds, num_models)

Arguments

preds

vector of predictions as obtained from xgb.train

number

of models

Value

preds in matrix form, with num_models columns and num_obs rows


reichlab/xgbstack documentation built on May 27, 2019, 4:54 a.m.