predict
-
ndarray item, [0] 차이점카테고리 없음 2023. 4. 24. 11:44
XGBoost 모델로 predict를 했는데 predict 결과가 array로 나온다. 예측값을 round 처리도 해주고, 다른 변수랑도 더해주고 해야하는데 array type이면 에러가 남. 그래서 array에서 값을 빼서 사용해야됨. 방법1. pred = model.predict(x_test) pred.item() 방법2. pred = model.predict(x_test) pred[0] 이 둘 작업을 한 후에 type을 확인해보면 float 형식 근데 값이 조금 다르다?! item 일 땐(방법1일 땐) 소숫점 아래로 많이 나오는데, [0] 일 때는(방법2일 땐) 어느정도 정리가 됨. 뭘 사용하는 게 적합한지는 모르겠지만, item으로 뽑을 때가 원본에 가까운 것 같아서 item으로 뽑은 후에 rou..
-
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 로 데이터를 추출을 ..