float
-
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..