A comprehensive toolkit for large model evaluation
Project description
👋 join us on Discord and WeChat
🧭 Welcome
to OpenCompass!
Just like a compass guides us on our journey, OpenCompass will guide you through the complex landscape of evaluating large language models. With its powerful algorithms and intuitive interface, OpenCompass makes it easy to assess the quality and effectiveness of your NLP models.
🔥 Attention
We launch the OpenCompass Collabration project, welcome to support diverse evaluation benchmarks into OpenCompass! Clike Issue for more information. Let's work together to build a more powerful OpenCompass toolkit!
🚀 What's New ![](https://pypi-camo.freetls.fastly.net/1c42fc0b389e97c73483a8548b48f208ebf9643b/68747470733a2f2f757365722d696d616765732e67697468756275736572636f6e74656e742e636f6d2f31323738323535382f3231323834383136312d35653738336464362d313165382d346665302d626262612d3339666662373737333062652e706e67)
- [2023.09.20] We update the leaderboard with InternLM-20B, welcome to our homepage for more details. 🔥🔥🔥.
- [2023.09.19] We update the leaderboard with WeMix-LLaMA2-70B/Phi-1.5-1.3B, welcome to our homepage for more details. 🔥🔥🔥.
- [2023.09.18] We have released long context evaluation guidance. 🔥🔥🔥.
- [2023.09.08] We update the leaderboard with Baichuan-2/Tigerbot-2/Vicuna-v1.5, welcome to our homepage for more details.
- [2023.09.06] Baichuan2 team adpots OpenCompass to evaluate their models systematically. We deeply appreciate the community's dedication to transparency and reproducibility in LLM evaluation.
- [2023.09.02] We have supported the evaluation of Qwen-VL in OpenCompass.
- [2023.08.25] TigerBot team adpots OpenCompass to evaluate their models systematically. We deeply appreciate the community's dedication to transparency and reproducibility in LLM evaluation.
- [2023.08.21] Lagent has been released, which is a lightweight framework for building LLM-based agents. We are working with Lagent team to support the evaluation of general tool-use capability, stay tuned!
✨ Introduction
OpenCompass is a one-stop platform for large model evaluation, aiming to provide a fair, open, and reproducible benchmark for large model evaluation. Its main features includes:
-
Comprehensive support for models and datasets: Pre-support for 20+ HuggingFace and API models, a model evaluation scheme of 50+ datasets with about 300,000 questions, comprehensively evaluating the capabilities of the models in five dimensions.
-
Efficient distributed evaluation: One line command to implement task division and distributed evaluation, completing the full evaluation of billion-scale models in just a few hours.
-
Diversified evaluation paradigms: Support for zero-shot, few-shot, and chain-of-thought evaluations, combined with standard or dialogue type prompt templates, to easily stimulate the maximum performance of various models.
-
Modular design with high extensibility: Want to add new models or datasets, customize an advanced task division strategy, or even support a new cluster management system? Everything about OpenCompass can be easily expanded!
-
Experiment management and reporting mechanism: Use config files to fully record each experiment, support real-time reporting of results.
📊 Leaderboard
We provide OpenCompass Leaderbaord for community to rank all public models and API models. If you would like to join the evaluation, please provide the model repository URL or a standard API interface to the email address opencompass@pjlab.org.cn
.
📖 Dataset Support
Language | Knowledge | Reasoning | Comprehensive Examination | Understanding |
Word Definition
Idiom Learning
Semantic Similarity
Coreference Resolution
Translation
|
Knowledge Question Answering
Multi-language Question Answering
|
Textual Entailment
Commonsense Reasoning
Mathematical Reasoning
Theorem Application
Code
Comprehensive Reasoning
|
Junior High, High School, University, Professional Examinations
|
Reading Comprehension
Content Summary
Content Analysis
|
📖 Model Support
Open-source Models | API Models |
|
|
🛠️ Installation
Below are the steps for quick installation and datasets preparation.
conda create --name opencompass python=3.10 pytorch torchvision pytorch-cuda -c nvidia -c pytorch -y
conda activate opencompass
git clone https://github.com/open-compass/opencompass opencompass
cd opencompass
pip install -e .
# Download dataset to data/ folder
wget https://github.com/open-compass/opencompass/releases/download/0.1.1/OpenCompassData.zip
unzip OpenCompassData.zip
Some third-party features, like Humaneval and Llama, may require additional steps to work properly, for detailed steps please refer to the Installation Guide.
🏗️ ️Evaluation
After ensuring that OpenCompass is installed correctly according to the above steps and the datasets are prepared, you can evaluate the performance of the LLaMA-7b model on the MMLU and C-Eval datasets using the following command:
python run.py --models hf_llama_7b --datasets mmlu_ppl ceval_ppl
OpenCompass has predefined configurations for many models and datasets. You can list all available model and dataset configurations using the tools.
# List all configurations
python tools/list_configs.py
# List all configurations related to llama and mmlu
python tools/list_configs.py llama mmlu
You can also evaluate other HuggingFace models via command line. Taking LLaMA-7b as an example:
python run.py --datasets ceval_ppl mmlu_ppl \
--hf-path huggyllama/llama-7b \ # HuggingFace model path
--model-kwargs device_map='auto' \ # Arguments for model construction
--tokenizer-kwargs padding_side='left' truncation='left' use_fast=False \ # Arguments for tokenizer construction
--max-out-len 100 \ # Maximum number of tokens generated
--max-seq-len 2048 \ # Maximum sequence length the model can accept
--batch-size 8 \ # Batch size
--no-batch-padding \ # Don't enable batch padding, infer through for loop to avoid performance loss
--num-gpus 1 # Number of required GPUs
Through the command line or configuration files, OpenCompass also supports evaluating APIs or custom models, as well as more diversified evaluation strategies. Please read the Quick Start to learn how to run an evaluation task.
🔜 Roadmap
- Subjective Evaluation
- Release CompassAreana
- Subjective evaluation dataset.
- Long-context
- Long-context evaluation with extensive datasets.
- Long-context leaderboard.
- Coding
- Coding evaluation leaderdboard.
- Non-python language evaluation service.
- Agent
- Support various agenet framework.
- Evaluation of tool use of the LLMs.
- Robustness
- Support various attack method
👷♂️ Contributing
We appreciate all contributions to improve OpenCompass. Please refer to the contributing guideline for the best practice.
🤝 Acknowledgements
Some code in this project is cited and modified from OpenICL.
Some datasets and prompt implementations are modified from chain-of-thought-hub and instruct-eval.
🖊️ Citation
@misc{2023opencompass,
title={OpenCompass: A Universal Evaluation Platform for Foundation Models},
author={OpenCompass Contributors},
howpublished = {\url{https://github.com/open-compass/opencompass}},
year={2023}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for opencompass-0.1.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f4aef26dd3acf43c7596c424abae2d10098f79c1988c287ddcf5d2cd1a1cb344 |
|
MD5 | 89807bdb19019e185fdb309bfde96e6a |
|
BLAKE2b-256 | 3077d83bbbf6c6c4a22e126eab3851b7f2ba0b968ffc29c78b170463879e31a9 |