> ## Documentation Index
> Fetch the complete documentation index at: https://docs.cosmosid.com/llms.txt
> Use this file to discover all available pages before exploring further.

# Overview

> How the ML Regression and Classification modules learn from microbiome data to predict a target, and when to use each.

## What is machine learning?

Machine learning builds a model that learns patterns from labelled examples, meaning samples where the outcome is already known, and uses those patterns to predict the outcome for new samples. This is supervised learning, where every training sample carries a known target. It differs from unsupervised methods such as clustering, which group samples by similarity without any predefined outcome.

On Cosmos-Hub 2.0, the inputs for each sample are its microbial **features** (the relative abundance of taxa or functions) together with any **covariates** (additional sample information such as sex, treatment, disease status, or BMI). The outcome the model learns to predict is the **target variable**.

The two modules differ in the **type of target** they predict:

* **ML Regression** predicts a **continuous value**, such as a metabolite concentration, an inflammation marker, or a clinical score.
* **ML Classification** predicts a **category**, such as patient versus healthy control, responder versus non-responder, or one of several body sites.

In short, regression answers *"how much?"* and classification answers *"which group?"*. Aside from the target type, the two modules share most of their workflow and parameters, which is why they are documented together. Once a model predicts accurately, it can also be inspected to rank which features contributed most, a result called **feature importance**.

## Why is it useful?

* **Predict a trait or class from the microbiome.** Build reusable models that estimate the target for unseen samples from their microbial features, for example whether a stool sample comes from a patient or a control, as commonly done in host-phenotype classification studies.
* **Find candidate feature to follow up.** Feature importance produces a data-driven shortlist of the taxa or functions that most drive accurate prediction, a common route to biomarker discovery.
* **Capture effects that linear models miss.** Flexible algorithms such as random forests or gradient boosting can model nonlinear relationships, for example a taxon that matters only above a threshold, or only in combination with another.

This module fits when you have a labelled dataset with a known outcome for each sample and want a model that generalizes to new samples. To simply ask which taxa differ between predefined groups, differential abundance testing is the more direct tool.

## How does it work?

<Steps>
  <Step title="Assemble the input">
    Each sample's microbial features in your Cohort's Data Table, your Target Variable that you want to predict, and  covariates are assembled into the input Data Table.
  </Step>

  <Step title="Train the model">
    The chosen algorithm learns the patterns that relate the features to the Target Variable on part of the data.
  </Step>

  <Step title="Evaluate with cross-validation">
    Performance is checked on held-out samples using K-fold cross-validation, providing an estimate of how well the model generalizes to new data.
  </Step>

  <Step title="Read predictions and feature importance">
    Results are summarized as accuracy diagnostics plus a ranked list of the features that most influenced the predictions.
  </Step>
</Steps>
