Abstract: A tax category prediction model, which is trained using a tax category prediction dataset, provides a tax category prediction for an item having an item listing. The tax category prediction model enhances the ability and accuracy of identifying a tax category associated with an item for sale via the online marketplace (e.g., identifying the tax category before the item is offered for sale via the online marketplace). The tax category prediction dataset is generated based on one or more of a text embedding, an image embedding, another type of embedding, or a combination thereof. The text embedding may be identified by applying a natural language processing model to a text string of the item listing. The natural language processing model may comprise bidirectional encoder representations from transformers (BERT).