AI-Based Recommendation Engine for Retail – SRM Capstone

AI-Based Recommendation Engine for Retail – SRM Capstone
Project thumbnail
Project thumbnail
Project thumbnail
Project Innovative

AI-Based Recommendation Engine for Retail – SRM Capstone

Developed a machine learning-based recommendation system for retail personalization using AWS and Python.

Kriti Iyer

Kriti Iyer

B.Tech Student | Machine Learning Explorer: Unleashing the Power of AI Innovation

259
Views
3
Claps
0
Comments

Project Overview

I I was part of her final year capstone project at SRM University, Kriti Iyer built an intelligent recommendation engine tailored for the retail industry. The objective I was to create a system that could provide personalized product recommendations based on user behavior, purchase history, and browsing patterns. Kriti employed machine learning algorithms including collaborative filtering and content-based filtering, using Python and AWS SageMaker for model training and deployment. The engine processed anonymized user data from a sample retail dataset to predict user preferences with high accuracy. What made her project stand out I was the seamless integration with a mock e-commerce frontend that demonstrated how real-time recommendations could enhance user engagement and sales. Kriti also implemented A/B testing to compare recommendation efficiency and optimize the model accordingly. The project emphasized the growing importance of AI in e-commerce and proved her capability to take an end-to-end ML project from concept to deployment. The success of the project led her team to present it at the SRM Annual Tech Symposium.

Project Claps

3 claps

Recent Clappers

Showing 3 of 3 clappers

Project Images

Discussion

Please log in to join the discussion.

More Projects You Might Like

Similar Projects

Skill Salary Correlation Study

Skill Salary Correlation Study

This project presents a comprehensive data analysis framework aimed at uncovering how technical skills, experience levels, job roles, and industry sectors influence salary outcomes within the data science and technology domains. Its central purpose is to help professionals understand which skills yield the highest compensation, how experience affects income growth, and which combinations of competencies offer the greatest market value. The framework implements a complete end-to-end data pipeline that efficiently transforms raw salary data into meaningful, data-driven insights. The process begins with data preparation, where raw CSV datasets containing salary and skill-related information are processed, cleaned, and standardized. The system can automatically import data from Kaggle or generate synthetic data if required. It further extracts key features, normalizes salary information, and prepares analysis-ready datasets. In the statistical modeling phase, the project employs regression-based techniques to quantify the impact of various skills, experience levels, and industry factors on compensation. These models assign measurable values to specific skills, providing an interpretable understanding of their influence on salary variations. The next stage, insight generation, produces detailed analytical reports, model summaries, and visual outputs including correlation matrices, salary-by-skill comparisons, and coefficient impact charts that highlight high-value skill sets. To make insights more accessible, the project incorporates an interactive dashboard built using Dash, which allows users to filter results dynamically by skill, job title, industry, or country and visualize customized insights through engaging, data-rich charts. The project’s structure reflects strong software engineering practices with a clear separation of modules—core data processing and modeling scripts reside in the src directory, visual components in the dashboard directory, datasets in the data folder, and reports in the reports section. Exploratory research and experimental notebooks are also available for deeper exploration. Designed for simplicity and extensibility, the framework allows one-command execution for data preparation, modeling, and dashboard deployment. It supports both sample and custom datasets with comprehensive documentation to assist users in expanding the analysis. Ultimately, this project serves as an intelligent career intelligence platform—empowering data professionals, recruiters, and job seekers to make informed, evidence-based decisions about skill development and salary expectations in the evolving technology landscape.

Akash Kumar Singh Akash Kumar Singh
Employee Attrition Prediction using XGBoost and Flask with SHAP-based Feature Insights

Employee Attrition Prediction using XGBoost and Flask with SHAP-based Feature Insights

This project focuses on predicting employee attrition — identifying whether an employee is likely to leave the organization — using XGBoost, a high-performance gradient boosting algorithm. The model is trained on an HR dataset and incorporates advanced feature selection to highlight the top factors influencing attrition. Key steps include data preprocessing, encoding categorical variables, feature importance extraction, and model training using XGBoost. The project emphasizes explainability through SHAP (SHapley Additive exPlanations) values, which visualize and rank the top 10 features impacting employee turnover, such as OverTime, JobLevel, MaritalStatus, and TotalWorkingYears. To make it interactive, the prediction system is deployed using Flask, allowing users to input employee details (such as job level, income, overtime status, etc.) and instantly receive an attrition prediction. The app also displays visual insights derived from SHAP, helping HR managers understand why a particular prediction was made. Tech Stack: Python, Flask, XGBoost, SHAP, Pandas, Scikit-learn, HTML/CSS Key Features: ML-based employee attrition prediction SHAP visual explanation for top 10 contributing factors Flask web interface for live predictions Clean, modular code structure for scalability This project demonstrates a complete data science pipeline — from data analysis and feature engineering to model deployment and visualization, aligning with real-world HR analytics use cases.

Miriyala Veera Ganesh Miriyala Veera Ganesh

More by  Kriti Iyer