Key Features
The following key features are currently a part of the UpTrain package (and we are constantly adding more). If you did not find what you were looking for, submit a feature request.
- Model performance monitoring and Concept Drift : UpTrain tracks the performance of a machine learning model over time, including metrics such as accuracy, precision, recall, and F1 score. It provides visualizations and alerts to help users understand how well the model is performing (the following images are from the ride-time estimation example)
![]() | ![]() |
---|---|
- Data Drift Detection : UpTrain uses advanced techniques such as statistical tests and change point detection algorithms to detect any changes in the distribution of data over time. This allows users to identify when their modelās performance might be negatively impacted by data drift and take appropriate action

Data drift detection through UpTrain in the human orientation classification example
- Edge Case Detection : UpTrain uses user-defined signals and statistical techniques such as outlier detection to identify data points that fall outside of the typical range of values. These edge cases can be challenging for a machine learning model to predict accurately, and UpTrain allows users to flag and handle these cases separately

Edge case detection from finetuning LLMs example
- Data Integrity Monitors : UpTrain checks for missing or inconsistent data, duplicate records, and other potential issues that could affect the accuracy of a machine learning model. It also checks for data quality issues such as outliers, missing values, and incorrect data types

Data integrity monitoring from the finetuning LLMs example
- Customizable Metrics : UpTrain allows users to add their own custom metrics to monitor, such as business-specific KPIs. These metrics can be easily added to the dashboard and used alongside other performance metrics to gain a more complete understanding of the modelās performance

A custom monitor from the Fraud Detection Example
- Quantifying Model Bias : Understand and quantify biases such as popularity, position, and selection biases. UpTrainās model bias detection feature identifies and addresses potential biases in machine learning models by analyzing training data, predicting outcomes, and comparing performance across demographics. This ensures models provide accurate and unbiased results, promoting responsible and ethical use of machine learning.

Histogram for popularity bias from Shopping Cart Recommendation Example
- t-SNE Dimensionality Reduction : Visualize embeddings by reducing their dimensionality using t-SNE.

t-SNE dimensionality reduction for the Text Summarization example
- Visualization using UMAP : Visualize embeddings using UMAP to see how posts cluster upon convergence.

UMAP visualization for the Text Summarization example
- AI Explainability : AI explainability refers to machine learning modelsā ability to explain their reasoning in a way humans can understand. It improves transparency and trust in machine learning systems, enabling users to understand the factors influencing a modelās predictions. It also helps understand the relative importance of each feature in predictions.

SHAP explainability of features from the Ride-time Estimation example
-
Smart data-point collection : UpTrain automatically collects data points that fall outside the typical range of values or that cause data drift for use in automated retraining of the model. This allows users to constantly improve the performance of their model by retraining it with new, relevant data
-
A/B testing : Live monitor your A/B tests with the UpTrain dashboard and generate insights in days (not months).

Simultaneous live monitoring of two models in UpTrain dashboard
-
Data Security : With UpTrain, your data never goes out of your machine. Many ML models operate on user-sensitive data, and may not always be possible to send usersā private data to third parties. This motivated UpTrain to create an open-source self-hosted alternative from a privacy perspective.
-
Slack Integration : Setting up alerts on Slack is easy with UpTrain. Users can simply configure their Slack workspace, select the metrics they want to monitor, and set the alert thresholds. When a modelās performance falls outside of the defined thresholds, UpTrain sends an alert to the userās designated Slack channel, enabling them to take immediate action.