A retail company intends to use machine learning to categorize new products. A labeled dataset of current products was provided to the Data Science team. The dataset includes 1,200 products. The labeled dataset has 15 features for each product such as title dimensions, weight, and price. Each product is labeled as belonging to one of six categories such as books, games, electronics, and movies.
Which model should be used for categorizing new products using the provided dataset for training?
- AnXGBoost model where the objective parameter is set to multi:softmax
- A deep convolutional neural network (CNN) with a softmax activation function for the last layer
- A regression forest where the number of trees is set equal to the number of product categories
- A DeepAR forecasting model based on a recurrent neural network (RNN)
Answer(s): A
Reference:
https://medium.com/@gabrielziegler3/multiclass-multilabel-classification-with-xgboost-66195e4d9f2d
Reveal Solution Next Question