//pragmatic leaders

Types of Machine Learning

Reading time
6 min
Section
PM Foundations (Legacy)
6 min left0%
types of machine learning0%
6 min left
Machine learning is not magic. It is a set of methods to help your product learn from data and improve over time.
Talvinder Singh, from a Pragmatic Leaders AI cohort

Machine learning is a powerful tool in product management — but it is not a single monolith. There are distinct types of machine learning, each with different assumptions, workflows, and use cases. Knowing which type you are dealing with is critical to setting realistic expectations and guiding your teams effectively.

Most machine learning applications fall into three broad categories: supervised learning, unsupervised learning, and reinforcement learning. Each type answers a different kind of product question and requires different data and design.

Supervised learning is task-driven and outcome-oriented

Supervised learning is the most common and intuitive type of machine learning. You start with a dataset where the inputs and the correct outputs are known. The algorithm learns a mapping from inputs to outputs by training on labeled examples. Once trained, it predicts outputs for new, unseen inputs.

The actual job of supervised learning is to build a predictor function that, given input data, estimates the correct output as accurately as possible. For example, predicting the price of a house based on its size, location, and other features.

The process has two phases:

  1. Training: The model sees many examples with input-output pairs, such as (5,6) → 11 if the task is addition. It learns the underlying logic or pattern.

  2. Prediction: The model applies the learned logic to new inputs, like (2,4) → 6.

Supervised learning is often fast and accurate when you have good labeled data.

Two main types of supervised learning: regression and classification

Supervised learning breaks down into two subtypes depending on the nature of the output you want to predict:

  • Regression: Predict continuous values. For example, forecasting the stock price tomorrow or estimating total runs in a cricket match based on past data.

  • Classification: Predict discrete categories or labels. For example, determining whether an email is spam or not, or whether it will rain today (yes/no).

Both rely on the same core idea: learning from labeled data to predict the right output for new inputs.

// scene:

Product team meeting at a Bangalore fintech startup

You (PM): “We have historical transaction data labeled as fraudulent or legitimate. Can we build a model to detect fraud?”

Data Scientist: “Yes, this is a classic supervised learning problem — classification. We'll train the model on past transactions and their labels.”

You (PM): “Great. How do we validate that the model is accurate enough for production?”

Data Scientist: “We'll measure precision, recall, and false positives on a test set. We want to minimize false alarms.”

You (PM): “Perfect. Let's set acceptance criteria based on those metrics and plan deployment.”

// tension:

Aligning product goals with supervised learning capabilities

Unsupervised learning is data-driven pattern discovery

Unsupervised learning is fundamentally different. Instead of labeled data, you give the algorithm raw input data without explicit outputs. The algorithm's job is to find structure, patterns, or groupings in the data on its own.

This type of learning is useful when you don’t know what the output should be — you want to explore the data to discover hidden relationships or clusters.

Common use cases:

  • Clustering: Grouping similar items together, like grouping tweets based on content or photos based on objects detected.

  • Dimensionality reduction: Simplifying data to its essential features.

Unsupervised learning is heavily dependent on the richness and diversity of the input data because it must infer everything from patterns alone.

// thread: #product-ml — Using unsupervised learning for customer insight
You (PM)We have millions of customer reviews but no labels. Can we use ML to find common themes?
Data AnalystYes, we can run clustering algorithms to group reviews by topic without any labeled data.
You (PM)Will this help us identify new pain points or feature requests?
Data AnalystExactly. It reveals patterns we might not have noticed.

Reinforcement learning learns by interaction and feedback

Reinforcement learning is a third type that models learning as a process of trial and error. An agent interacts with an environment, performs actions, and receives rewards or penalties. Over time, it learns a policy to maximize cumulative rewards.

This is less common in product management but critical in domains like robotics, game AI, and autonomous systems.

An intuitive example:

  • A child approaches a fireplace (the environment).
  • Feeling warmth (+1 reward) is positive.
  • Touching fire and getting burned (-1 penalty) is negative.
  • The child learns to avoid touching the fire but to stay near for warmth.

Reinforcement learning algorithms balance exploration (trying new actions) and exploitation (using known good actions).

When to use each type in product management

TypeWhat it doesData neededTypical product use casesIndian example
SupervisedLearn to predict outputs from inputsLabeled data (input-output pairs)Fraud detection, churn prediction, credit scoringRazorpay fraud models
UnsupervisedDiscover patterns or clusters in dataUnlabeled dataCustomer segmentation, recommendation engines, anomaly detectionMeesho user clustering
ReinforcementLearn optimal actions through feedbackInteraction data with rewardsPersonalization engines, game AI, dynamic pricingSwiggy delivery route optimization (experimental)

In practice, supervised learning is the starting point for most business problems because labeled data is often available or can be created. Unsupervised learning complements this by surfacing new insights from raw data. Reinforcement learning is specialized and requires careful setup.

The training-prediction cycle in supervised learning

Supervised learning projects follow a predictable flow:

  1. Data collection: Gather historical labeled data.

  2. Training: Use the data to build a predictive model.

  3. Validation: Test the model on new data to check accuracy.

  4. Deployment: Integrate the model into the product pipeline.

  5. Monitoring: Continuously measure performance and retrain as needed.

If you cannot answer how your model will be trained and evaluated before you build it, you are not ready to ship.

The pitfalls: data quality and bias

Machine learning is only as good as the data it learns from. In India, data quality is often messy:

  • Inconsistent formats.

  • Missing labels.

  • Multilingual text with code-switching.

  • Biases that reflect social inequalities.

These challenges require extra effort in data cleaning, feature engineering, and fairness audits before trusting model outputs.

Tools for machine learning development

There are many tools available to build ML models. Some popular ones include:

  • TensorFlow (Google): Open-source library for building ML models.

  • PyTorch (Facebook): Flexible deep learning framework.

  • AWS SageMaker: Managed service to build, train, and deploy models.

  • Google Cloud AI: Suite of ML services and APIs.

Choosing the right tool depends on your team's skills, infrastructure, and product needs.

Field exercise: Identify ML types in your product

Take any product you use regularly — Swiggy, Razorpay, Meesho, or your own startup’s product — and answer these:

  1. What is a problem the product solves that involves predicting or classifying?

  2. Is there labeled data available to train a supervised model?

  3. Are there cases where the product discovers patterns without explicit labels?

  4. Could the product benefit from a system that learns from its own interactions?

Write down which ML type best fits each case and why.

// exercise: · 15 min
Classify ML types in your product

Pick one product you know well. For each of these questions, write a short answer:

  1. What supervised learning problems exist? (e.g., fraud detection, churn prediction)

  2. What unsupervised learning problems exist? (e.g., user segmentation, anomaly detection)

  3. Is reinforcement learning applicable? Why or why not?

  4. What data challenges might impact ML success?

This exercise grounds ML types in your real-world context.

Test yourself: Choosing the right ML approach

// learn the judgment

You are PM at a Series B fintech startup in Mumbai. The data team proposes two projects: (1) a supervised model to predict loan default using labeled repayment data, and (2) an unsupervised clustering of transaction histories to identify new customer segments. Your CEO wants to prioritize one for next quarter.

The call: Which project do you prioritize and why? How do you justify your choice to the CEO and engineering leads?

Your reasoning:

// practice

You are PM at a Series B fintech startup in Mumbai. The data team proposes two projects: (1) a supervised model to predict loan default using labeled repayment data, and (2) an unsupervised clustering of transaction histories to identify new customer segments. Your CEO wants to prioritize one for next quarter.

Your task: Which project do you prioritize and why? How do you justify your choice to the CEO and engineering leads?

your reasoning:

0 chars (min 80)

From the field: ML misconceptions in Indian startups

Where to go next

PL alumni now work at Razorpay, Meesho, Swiggy, Flipkart, and other leading Indian tech companies.