Back To Projects

Predicting Investment Success on Shark Tank India: A Machine Learning Approach

Jia A.

This project explores the use of machine learning—specifically regression models, neural networks, and transfer learning—to predict the success of startup pitches on Shark Tank, with a focus on bridging insights between U.S. and Indian markets. By addressing the unique challenges faced by Indian entrepreneurs, the project demonstrates how advanced modeling can uncover patterns in investment decisions and support funding opportunities in emerging economies.


Shark Tank is a critical platform for emerging entrepreneurs, offering not only financial investment but also valuable visibility from renowned investors, or "Sharks." While extensive research has been conducted on Shark Tank in Western markets, there is a notable gap in studies focusing on the Indian context. India has one of the fastest-growing economies in the world, with a rapidly expanding startup ecosystem that is significantly different from Western markets. Additionally, Indian entrepreneurs face unique challenges, such as limited access to funding, regional disparities, and a complex regulatory environment, which make the Shark Tank platform particularly impactful for their success. Understanding the specific factors that influence investment decisions in this context—such as investor preferences, market fit, and demographic considerations—can provide invaluable insights for entrepreneurs looking to secure funding. This study addresses the challenge of predicting whether a pitch will receive an offer by leveraging a diverse dataset with input factors from both U.S. and Indian datasets. It employs a range of regression models, a neural network, and transfer learning techniques to adapt insights from U.S. data to the Indian context, improving the model's predictive accuracy across different markets. Results indicate that the neural network model achieved the highest predictive accuracy, capturing complex interactions better than simpler models. Remarkably, transfer learning provided reasonable results even after removing city and state features, underscoring its adaptability across diverse markets. These findings highlight the potential of advanced machine learning techniques to improve understanding of investment dynamics in emerging economies like India.

Explore More!

Jia A.
Shreyaa Raghavan

Related Projects

Self-supervised DINOv2 versus Supervised CNNs for Insurance Claim Cost Predictions

This study investigates whether vehicle insurance claim amounts can be predicted directly from images of damaged cars by comparing CNN models, the self-supervised DINOv2 model, and GPT-5.5 as a vision language model. Results showed that DINOv2 performed best, achieving an R² of 0.12 compared with 0.07 for ResNet152V2 and 0.08 for ResNeXt50, while GPT-5.5 tended to overestimate repair costs, particularly for severe damage.
Rohan V.
Mentored by Ivan Felipe Rodriguez
Approaches to fraud detection on credit card transactions using artificial intelligence methods

In this paper, we study the problem of detecting fraudulent credit card transactions. We select the most relevant features using a heuristic approach, and fit three different model classes to a simulated dataset: Logistic Regression, Random Forests and Gradient Boosting Machines. We find that hyperparameter tuning has a big impact on the precision and recall of our classifiers. We also find that of the three classes, Gradient Boosting Machines were the best-performing model class, achieving 83% precision and 64% recall on unseen data.
Aryaman R.
Mentored by Yuan Lee
Predicting Repeat Purchases in E-Commerce

The underlying problem here is identifying more efficient upselling strategies for retailer companies. These findings may potentially help a retail company determine the likelihood of a customer purchasing another item based on their first purchase, and then decide how to upsell based on that information.
Ayrton S.
Mentored by Bryce Johnson