Neighbors-based prediction of physical function after total knee arthroplasty