Method, article of manufacture, and system for providing category suggestions

Abstract

A method for categorizing a product, the method including receiving information for the product; inputting the information into a trained machine learning model for a taxonomy tree; receiving a plurality of arrays, each array representing a level in the taxonomy tree and consisting of probabilities for each category represented in the level that the product is categorized in that category; choosing, from a highest level tier array, a category having a highest probability, thereby designating a tier prediction; collecting, from a second level tier array, all children of the tier prediction; determining whether a highest probability from the children of the tier prediction exceeds a threshold, and if yes, choosing the category with the highest probably as a new tier prediction; and repeating the determining; when the threshold is not exceeded or if the tier prediction has no children, and selecting the tier prediction as a predicted category.