Capstone Projects
Build real-world ML applications across computer vision, NLP, and more
Plant Disease Detection
Build a CNN to classify plant leaf diseases for agricultural applications
PlantVillage Dataset (54K images)
Sentiment Analysis API
Create a sentiment analysis system with BERT and deploy as REST API
IMDB Reviews (50K reviews)
Customer Churn Prediction
Predict customer churn using ensemble methods with explainability
Telco Customer Churn (7K rows)
Stock Price Forecasting
Time series forecasting with LSTM and Transformer models
Yahoo Finance API (5 years data)
Real-time Object Detection
Build a YOLO-based object detector with video stream support
COCO Dataset (330K images)
Conversational AI Chatbot
Build a context-aware chatbot using RAG and LangChain
Custom Knowledge Base
Credit Card Fraud Detection
Detect fraudulent transactions with imbalanced learning techniques
Kaggle Credit Card Fraud (284K transactions)
Document Summarization
Fine-tune T5 for abstractive text summarization
CNN/DailyMail (300K articles)
Medical Image Segmentation
Segment tumors in MRI scans using U-Net architecture
BraTS Challenge (MRI scans)
Movie Recommendation System
Build a hybrid recommender with collaborative and content-based filtering
MovieLens 25M (25M ratings)
Game-Playing AI Agent
Train a DQN agent to play Atari games from pixels
Atari Environment (Gym)
Lane Detection System
Build a lane detection system for autonomous vehicles
TuSimple Lane Detection (6K images)