ndarray item, [0] 차이점

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으로 뽑은 후에 round로 정리해주는 방법을 사용했다.