Design an item category prediction system | Walmart Labs Interview Question