Walmart and Random Forest

Posted on Dec 7, 2015

Contributed by . He took NYC Data Science Academy 12 week full time Data Science Bootcamp program between Sept 23 to Dec 18, 2015. The post was based on his fourth class project(due at 8th week of the program).


My Github

In the recent Walmart Kaggle competition I used a Random Forest classifier to solve a market basket problem.  A market basket model is built on the idea there exists relationships between items purchased together.  For example, a person purchasing a new toothbrush is more likely to also purchase toothpaste than motor oil in the same shopping.  Retailers use these market basket relationships in the design of their stores for ease of use and also to increase sales.  In this specific problem Walmart has broken up their shopping trips into 38 unique 'TripType'.  They give you a training data set with over 600,000 rows and a test data set of the same size.  The training data set contained the 'TripType' and it was your job to predict the 'TripType' of the test set.  The 'TripTypes' were numeric dummy variables and contain no explicit information.  Each row in the data sets contained features about one item purchased.  Participants were given a 'Upc' code, a 'Weekday', 'DepartmentDescription', 'ScanCount', and 'VisitNumber'.  You are also given a 'FinelineNumber' which was used as internal market basket classifier for an item.  Unlike a Upc number which was unique to an item, the 'FinelineNumber' was shared by items commonly purchased together.  Toothpaste and toothbrushes might share a 'FinelineNumber'.  Participants were not allowed to use any external data in this contest.  There was plenty of information to extract from the data already given.  In the Walmart Kaggle forum, the person at the top of the leader board made a post stating that the secret to success in this contest would hinge on feature engineering and not so much on complicated ensemble methods.  Based on that advice, I leapt directly into generating new features.

I looked at the columns in the data set and thought about what information was there but not explicitly stated.  Feature engineering is also an example of where domain knowledge is important.  I thought about shopping trips and what are the things that make trips different.  Trips are different in their composition and quantity of items.  Through some very simple arithmetic operations one would be able to generate other features that improves the granularity of the dataset.  They tell you that any item that has a negative 'Scan Count' is a returned item.  Could that be an important feature?  Because you have the 'Scan Count' and the 'VisitNumber' you are able to calculate both the total number of items purchased and also the percentage of each trip that belongs to a specific item.  Once you have the total number of items in each trip, you are also able to generate features to show the percentage of each trip that belongs to a 'FinelineNumber' and a 'DepartmentDescription'.  It also should be noted that there were some missing data in both the training and test set.  I replaced the NaNs with a proxy value not found in the data set.  If you try deleting the rows based on the logic that they comprise such a small percentage of the data set, you will run into problems with your test submission, Kaggle expects the submission to have certain dimensions.   My code for this feature engineering is shown below.

Returned = grpwall.merge(pd.DataFrame({'ItemReturned':grpwall.ScanCount <= 0}),\
left_index = True, right_index = True)

Returned['RealSC'] = Returned.ScanCount

containsReturned = Returned.merge(pd.DataFrame\
left_on=['VisitNumber'], right_index=True)

NumberReturned = containsReturned.merge(pd.DataFrame\
left_on=['VisitNumber'], right_index=True)

nUnqItems = NumberReturned.merge(pd.DataFrame\
left_on=['VisitNumber'], right_index=True)

addFeats = nUnqItems.merge(pd.DataFrame\
left_on=['VisitNumber'], right_index=True)

grpUpc = pd.DataFrame({'percentTrip':addFeats.groupby\
prcntTrip = addFeats.merge(grpUpc, left_on=['VisitNumber', 'Upc'], right_index = True)

grpLineItem = pd.DataFrame({'percentLineTrip':addFeats.groupby\
prcntTripII = prcntTrip.merge(grpLineItem, left_on=['VisitNumber', 'FinelineNumber'], right_index = True)

grpDepartItem = pd.DataFrame({'percentDepartTrip':addFeats.groupby\
prcntTripIII = prcntTripII.merge(grpDepartItem,\
left_on=['VisitNumber', 'DepartmentDescription'], right_index = True)

Once you have your new features it's time to run some models.  I chose Random Forest due to it's ability to handle the large amounts of categorical features contained in the data set.  It also avoids the issue of over-fitting the data based on the way that the trees are built.  Each tree is built on a random subset of features that reduces the correlation between trees.  There is a beautiful library for Python called Scikit-learn that will have you building random forests in no time.  I split my training set in a 80/20 split, saving the 20% as a validation to get an estimate of the test error rate.  I then built my model on the entire training set and then made a prediction with my test set.  I ran Random Forest a number of times as I engineered different features and you can see the results of my Kaggle submissions below.  Kaggle uses logloss as a measure of accuracy, it can be thought of as the amount of entropy attached to a prediction.

It should be noted that the features that I mentioned above were not the only features that I tried and you can see from my Kaggle results not every feature adds to the predictive power of your model.  Once I had features that I was happy with, it was time to tune Random Forest.  While just making submissions based on each new feature that improved my estimate of the test error was a reasonable process for feature engineering, it wouldn't work for parameter tuning.  There are way too many possible parameter options to use such an inefficient process.  The best way to tune your parameters is to use a grid search with cross validation.  In this process you create a matrix of all ranges of values for your parameters.  You then run Random Forest on all of these possible parameter values and record the estimated accuracy.  This will give you the best combination of parameter values.  You used cross validation because multiple splits will give you a better estimate of the test error than one split alone.  I used code directly from the Scikit-learn documentation for this process.  This method gave me the optimized parameters that provided my best score.  I'm continuing to work on the Walmart problem but with two advancements.  I'm trying to see if XGBoost gives me better results than Random Forest (not yet) and I'm also moving my work on to an AWS instance to improve speed.  If you want to see what I did in its entirety please check out my Github link at the top of the page.

About Author

William Aiken

Nate Aiken graduated from City College in 2014 with a BS in Biology with a focus in Neuroscience. His experience studying vision and hearing in labs at City and Rockefeller University led him to the bootcamp. He enjoys...
View all posts by William Aiken >

Leave a Comment

No comments found.

View Posts by Categories

Our Recent Popular Posts

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI