MetricWhiz is a command-line application designed to process prediction files and provide a comprehensive evaluation of binary classification models. The application offers an interactive terminal interface to visualize metrics, adjust thresholds, and explore classification results.
- Command-Line Interface: Easily specify the path to your prediction file.
- Interactive Terminal Interface: Navigate through different tabs to view reports and sample data.
- Threshold Adjustment: Dynamically adjust the classification threshold to see its impact on metrics.
- Classification Metrics: View precision, recall, F1 score, accuracy, specificity, MCC, and AUROC.
- Confusion Matrix: Visual representation of true positives, true negatives, false positives, and false negatives.
- Sample Exploration: Filter and examine samples by their classification category.
To use MetricWhiz, ensure you have Rust installed on your system. Then, clone the repository and build the application:
git clone <repository-url>
cd metricwhiz
cargo build --release
Pre-built binaries for MetricWhiz are available for Windows, macOS, and Linux. You can download them from the Releases page on GitHub. Follow these steps:
- Navigate to the Releases page.
- Find the latest release.
- Download the appropriate binary for your operating system:
metricwhiz-windows.exe
for Windowsmetricwhiz-macos
for macOSmetricwhiz-linux
for Linux
- Make sure the binary is executable (on macOS and Linux, you may need to run
chmod +x <binary-name>
).
To run MetricWhiz, use the following command:
./metricwhiz <path-to-prediction-file>
Arguments:
<path-to-prediction-file>
: The path to the CSV file containing predictions. The file must includepred_score
,ground_truth
, andtext
columns.
- 'q': Quit the application.
- 'n': Move to the next tab.
- 'p': Move to the previous tab.
- Left/Right Arrow Keys: Adjust the classification threshold in the Report Viewer tab or navigate categories in the Sample Viewer tab.
- Up/Down Arrow Keys: Navigate through samples in the Sample Viewer tab.
- Report Viewer: Displays classification metrics, confusion matrix, and optimal thresholds.
- Sample Viewer: Allows exploration of individual samples categorized as True Positive, True Negative, False Positive, or False Negative.
If the specified prediction file does not exist, the application will terminate with an error message.
MetricWhiz is developed by Manpreet Singh.
This project is licensed under the Apache License 2.0 - see the LICENSE file for details.