Number of columns doesn’t match when predict
-
XGBoost predict 에러Error 2023. 3. 20. 18:38
Train data로 학습한 XGBoost 모델에 한 행만 input하여 predict를 하려고 하는데, 에러 발생 에러 메세지 XGBoostError: [18:30:57] ../src/predictor/cpu_predictor.cc:310: Check failed: m->NumColumns() == model.learner_model_param->num_feature (1 vs. 3) : Number of columns in data must equal to trained model. 에러 난 이유 train은 2차원 데이터로 학습을 했는데, 내가 한 행만 추출해서 input으로 넣으려니까 shape이 1차원이 됐고, 1차원 데이터를 넣으니까 위 에러가 났던 것이다. 그래서 0:1 로 데이터를 추출을 ..