Advanced E-Commerce App with GNN and AI for Inventory PredictionAdvanced E-Commerce App with GNN and AI for Inventory Prediction
The network for creativity
Join 1.25M professional creatives like you
Connect with clients, get discovered, and run your business 100% commission-free
Creatives on Contra have earned over $150M and we are just getting started
Back to feedPost
Karan's avatar
Karan Singh
• Jul 3
1. What it is: An advanced e-commerce web app that combines Graph Neural Networks (GNNs) and Generative AI to identify "toxic inventory"—products, sizes, or suppliers driving high customer returns.
2. GNN Predictive Modeling: Built a heterogeneous GNN using PyTorch Geometric to model relationships between customer demographics and product features to predict future return risks.
3. AI Attribute Enrichment: Automated a metadata extraction pipeline using the Gemini API with local caching to parse raw HTML product descriptions into structured product features (fabric, fit, pattern).
4. Interactive Dashboard & Reporting: Developed a Streamlit dashboard with Supabase Auth and integrated ReportLab to generate boardroom-ready PDF return audit reports.
5. Tech Stack: Python, PyTorch Geometric, Streamlit, Gemini API, Supabase, ReportLab.
Completed work
Data EngineeringMachine Learning EngineeringAI DevelopmentPythonPyTorchSupabase
Post image
Post image
Post image
Post image
Back to feed
The network for creativity
Join 1.25M professional creatives like you
Connect with clients, get discovered, and run your business 100% commission-free
Creatives on Contra have earned over $150M and we are just getting started